Jerome Rony 3491506fec Add foreach option for faster EMA

2022-11-30 14:06:58 -05:00

..

data

cast env var args for dataset readers to int

2022-10-17 14:40:11 -07:00

loss

Remove inplace operators when calculating the loss

2022-10-17 10:41:21 +02:00

models

refactor push_to_hub helper

2022-11-16 12:03:34 +01:00

optim

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

Add foreach option for faster EMA

2022-11-30 14:06:58 -05:00

__init__.py

…

version.py

Version 0.7.dev0 for master

2022-10-14 09:38:02 -07:00