Skip to content
Snippets Groups Projects
Commit df66e72e authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.segmentation.config.models.lwnet] Set default cosine annealing strategy...

[libs.segmentation.config.models.lwnet] Set default cosine annealing strategy to be closer to original publication
parent e70da2f1
No related branches found
No related tags found
1 merge request!46Create common library
Pipeline #89292 failed
......@@ -17,7 +17,9 @@ from torch.optim.lr_scheduler import CosineAnnealingLR
max_lr = 0.01 # start
min_lr = 1e-08 # valley
cycle = 50 # epochs for a complete scheduling cycle
# Original strategy: https://github.com/agaldran/lwnet/blob/master/train_cyclical.py#L298
# About 20 * len(train-data-loader)
cycle = 100 # 1/2 epochs for a complete scheduling cycle
resize_transform = ResizeMaxSide(512)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment