Anton Bakhtin fea7cc87f6 Move model to cuda before creating optimizer (#554)

2019-11-26 22:35:38 -05:00

..

__init__.py

split trainer mixins (#209)

2019-09-06 14:11:07 -04:00

amp_mixin.py

change print to logging (#457)

2019-11-05 08:43:21 -05:00

callback_config_mixin.py

Fix ModelCheckpoint default paths (#413)

2019-11-05 10:41:59 -05:00

data_loading_mixin.py

fix for pyTorch 1.2 (#549)

2019-11-26 10:58:50 -05:00

ddp_mixin.py

Release (#467)

2019-11-06 14:34:50 -05:00

dp_mixin.py

Move model to cuda before creating optimizer (#554)

2019-11-26 22:35:38 -05:00

evaluation_loop_mixin.py

Write progress bar to stdout (#531)

2019-11-21 13:26:24 -05:00

ignored_warnings.py

Dim 0 warning (#256)

2019-09-26 13:20:54 -04:00

logging_mixin.py

Ddp2 fix (#448)

2019-11-05 10:01:52 -05:00

model_hooks_mixin.py

Refactor (#407)

2019-10-22 04:16:51 +03:00

train_loop_mixin.py

Copy batch for local forward (#532)

2019-11-23 04:04:40 -05:00

trainer_io.py

change print to logging (#457)

2019-11-05 08:43:21 -05:00

trainer.py

Write progress bar to stdout (#531)

2019-11-21 13:26:24 -05:00

training_tricks_mixin.py

change print to logging (#457)

2019-11-05 08:43:21 -05:00