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 |
-
-
Save kagermanov27/269853a75433424fb05e09f542dbd5b2 to your computer and use it in GitHub Desktop.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment