Ross Wightman 6441e9cc1b Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.

2020-05-22 16:16:45 -07:00

..

__init__.py

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

cosine_lr.py

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00

plateau_lr.py

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00

scheduler_factory.py

Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.

2020-05-22 16:16:45 -07:00

scheduler.py

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00

step_lr.py

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00

tanh_lr.py

Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.

2020-02-22 16:23:15 -08:00