This repo contains files for CS5260 assignment 6.
Files are mapped as follows:
File Name | Scheduler | Learning Rate |
---|---|---|
Colossalai_lr_range_test.ipynb | LambdaLR | 0.1 |
Colossalai_lr_range_test_MultiStep.ipynb | MultiStep | 0.1158 |
Colossalai_lr_range_test_OneCycle.ipynb | OneCycle | 0.1158 |
Colossalai_lr_range_test_OneCycle_v2.ipynb | OneCycle | 2.05e-3 |
Colossalai_lr_range_test_OneCycle_v3.ipynb | OneCycle | 0.04128 |
Colossalai_lr_range_test_OneCycle_v4.ipynb | OneCycle | 0.2389 |