Dan Campbell 981758fd04 Only call load_spawn_weights if COLAB_GPU or KAGGLE_URL_BASE ...

environment variables are set

2020-04-27 19:46:29 -04:00

..

__init__.py

fixed new amp bugs (#1593)

2020-04-24 09:29:39 -04:00

auto_mix_precision.py

support for native amp (#1561)

2020-04-23 14:47:08 -04:00

callback_config.py

Progress bar callback (#1450)

2020-04-23 20:46:18 -04:00

callback_hook.py

Progress bar callback (#1450)

2020-04-23 20:46:18 -04:00

data_loading.py

ddp pickle

2020-04-27 07:19:12 -04:00

deprecated_api.py

Progress bar callback (#1450)

2020-04-23 20:46:18 -04:00

distrib_data_parallel.py

model checkpint on rank_zero_only & global rank state (#1408)

2020-04-24 17:21:00 -04:00

distrib_parts.py

model checkpint on rank_zero_only & global rank state (#1408)

2020-04-24 17:21:00 -04:00

evaluation_loop.py

Progress bar callback (#1450)

2020-04-23 20:46:18 -04:00

ignored_warnings.py

Dim 0 warning (#256)

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

logging.py

Clean up Argparse interface with trainer (#1606)

2020-04-26 09:20:06 -04:00

lr_finder.py

fix depreated call (#1596)

2020-04-24 14:45:43 -04:00

model_hooks.py

Remove error when test dataloader used in test (#1495)

2020-04-15 22:16:40 -04:00

optimizers.py

Fixed configure optimizer from dict without "scheduler" key (#1443)

2020-04-10 11:43:06 -04:00

supporters.py

Fixed configure optimizer from dict without "scheduler" key (#1443)

2020-04-10 11:43:06 -04:00

trainer.py

Only call load_spawn_weights if COLAB_GPU or KAGGLE_URL_BASE

2020-04-27 19:46:29 -04:00

training_io.py

ddp pickle

2020-04-27 06:51:42 -04:00

training_loop.py

Remove warning

2020-04-27 11:41:03 +02:00

training_tricks.py

support for native amp (#1561)

2020-04-23 14:47:08 -04:00