AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

LR scheduler + Resolution pb ResNet50/WRN

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-12 13:43:44 -05:00
parent 383f63c7b8
commit 79de0191a8
4 changed files with 78 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/smart_aug/dataug.py
View file

@ -958,6 +958,8 @@ class Augmented_model(nn.Module):
model.step(loss)
Does not support LR scheduler.
See ''run_simple_smartaug'' for a complete example.
Args: