Tullie Murrell
55edf7c922
Remove unneeded filename print ( #540 )
2019-11-23 04:00:39 -05:00
Tanel Alumäe
539d7bcb44
Avoid race condition in creating checkpoint directories ( #530 )
...
* Avoid race condition in creating checkpoint directories
In multi-GPU training, several processes run the code that creates checkpoint dirs. This fix avoids a probably rare situation (but it happened to me) where another process created a dir between the `exists` check and the `makedirs` call.
* Remove the now unneeded check for dir existence
2019-11-21 13:27:39 -05:00
Ir1dXD
7324dd902b
change Checkpoint callback's save_best_only to save_top_k ( #128 )
...
* docs: enable syntax highlight
* feat: change Checkpoint callback's `save_best_only` to `save_top_k`
fix #70
* docs: update docs for save_top_k
* revert other files
* style: lint for travis-ci
* fix typo
* make flake8 happy
* update according to review
* add tests
* rename func to private
* add doc on `save_top_k == 0`
* make flake8 happy
* update according to PR comments
* change some f-strings
* Update pt_callbacks.py
* Update test_models.py
* update options
* create folders
* Update test_models.py
* change epoch num
* support calling multiple times, add docs and tests
* update docs
* roll back changes in earlystopping
* clean test files
* make flake8 happy
* fix epoch number
* update tests about epoch numbers
* clean debugging code
* fix testing utils codes
* fix testing utils codes
* fix testing utils codes
* fix testing utils codes
* change save_dir to tests/tests according to previous lines
* remove unused overwrite option
* make flake8 happy
* change var name as per review
* make flake8 happy
* update property name to work on master
* elaborate in the docs
* update docs as per review
* revert previous commit
accidentally pressed wrong button when solving conflicts
2019-11-19 15:43:34 -08:00
Anton Bakhtin
e22dea228f
Remove errnonoeous comma in logging call ( #474 )
2019-11-08 08:50:39 -05:00
Nic Eggert
9fa2806605
Fix ModelCheckpoint default paths ( #413 )
...
* Make name and version properties required
* Warn before deleting files in checkpoint directory
* Get default checkpoint path from any logger
* Fix typos
* Uncomment logger tests
* Whitespace
* Update callback_config_mixin.py
checkpoints and version file names would just have a number. it's easy to tell what you're looking at with version_ prepended
* Address comments
* Fix broken tests
2019-11-05 10:41:59 -05:00
Ir1dXD
5a9afb11cc
change print to logging ( #457 )
...
* change print to logging
* always use logging.info
* use f-strings
* update code style
* set logging configs
* remove unused code
2019-11-05 08:43:21 -05:00
Jirka Borovec
f18aee30a5
Minor imports cleaning ( #402 )
...
* code cleaning
* drop unused imports
* optimize imports
2019-10-22 11:32:40 +03:00
William Falcon
416187fd56
Fixes #380
2019-10-18 15:33:00 +02:00
William Falcon
dcaba55251
Early stopping ( #332 )
...
* callbacks use all other keys in return dict
* callbacks use all other keys in return dict
* callbacks use all other keys in return dict
* callbacks use all other keys in return dict
* remove os.exit from early stopping
2019-10-08 16:21:00 -04:00
William Falcon
7099f8dbfb
split trainer mixins ( #209 )
...
* split trainer mixins
* Update multi_node_cluster_template.py
* Update single_cpu_template.py
* Update single_gpu_node_16bit_template.py
* Update single_gpu_node_ddp_template.py
* Update single_gpu_node_dp_template.py
* Update trainer_cpu_template.py
* Update trainer_io.py
* split trainer mixins
* Update multi_node_cluster_template.py
* deconflicted
* deconflicted
* deconflicted
2019-09-06 14:11:07 -04:00
Stanislav
73cf47112e
Gradient accumulation callback ( #150 )
...
* Gradient accumulation callback
* little test case
* typo
* import fix
* method name fix
* fix epochs indexing from 1
* better code style
* code style fix v2 :/
* change interface
* fix Trainre new api in tests
* trainer api bug fix
* new raising error, new update method
* extentions tests
* a little better tests
* typo fix
* flack8 better
* using scheduler for int and dict
* typo
* firs epoch bug fix
* test update
* empty dict exception
* floats check
* codestyle fix
* grad counting test
* someday, i will install normal linter
* add more checks
* Update test_models.py
* Update test_models.py
* Update test_models.py
* Update test_models.py
* Update test_models.py
* Update test_models.py
* Update test_models.py
2019-08-30 10:56:14 -04:00
William Falcon
10e4b18452
made imports absolute
2019-08-07 10:14:59 -04:00
Jiri BOROVEC
c44966a8bf
apply PEP8
2019-08-06 22:45:27 +02:00
Jiri BOROVEC
469941a528
pkg relative imports
...
* split requirements.txt
* pytest verbose
2019-08-05 10:52:09 +02:00
William Falcon
ffe0cd5965
removing unused imports
2019-08-04 13:08:45 -05:00
William Falcon
522af58504
clean up dead code
2019-07-03 17:05:20 -04:00
William Falcon
63d84283a4
removed checkpoint save_function option
2019-06-28 17:14:18 -04:00
William Falcon
4f75515ca4
adding docs
2019-06-27 11:04:02 -04:00
William Falcon
39af973bd4
added trainer docs
2019-06-27 11:03:53 -04:00
Tullie Murrell