Skip to content

Instantly share code, notes, and snippets.

@kagermanov27
Created June 2, 2022 18:24
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save kagermanov27/269853a75433424fb05e09f542dbd5b2 to your computer and use it in GitHub Desktop.
Save kagermanov27/269853a75433424fb05e09f542dbd5b2 to your computer and use it in GitHub Desktop.
Key Explanation
model_name Output model name without extension
criterion Criterion name for the training process
annotations_file Training file without _train, and extension in it
optimizer Optimizer name
lr Learning rate of optimizer
momentum Momentum of Optimizer
batch_size Number of items fetched in Custom Dataloader at each batch
n_epoch Number of epochs to run over training files
n_labels Number of labels to be trained, automatically gathered in another class
image_height Desired fixed image height
image_width Desired fixed image width
transform Whether or not input transformation should be applied
target_transform Whether or not label transformation should be applied
shuffle Whether or not Dataloader should shuffle the dataset to get a new item
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment