a6e7aa7796
allow using apex with any PT version ( #2865 )
...
* wip
* setup
* type
* name
* wip
* docs
* imports
* fix if
* fix if
* use_amp
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* fix tests
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* fix tests
* todos
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2020-08-08 11:07:32 +02:00
b7d72706c3
clean imports ( #2867 )
...
* clean imports
* miss
2020-08-08 00:33:51 +02:00
62ce00f96c
EvalResult support for val loop (PR 3/5) ( #2651 )
...
* add EvalResult to support to val/test loops
2020-07-22 13:53:10 -04:00
458bbad550
Avoid zeros in dice and iou ( #2567 )
...
* nones
* fix
* fix
* test
* test
* test
* fix
* eps
* tpu
* eps
* type
* test tpu
* Update __init__.py
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-07-09 20:40:10 -04:00
0be78d13aa
native amp ( #2373 )
...
* native amp
* typo
* imports
* apex
2020-06-26 21:45:13 -04:00
9df2b2090d
sets default ddp mode to spawn ( #2168 )
...
* set ddp_spawn as default
* spawn message
* spawn message
* spawn message
* spawn message
* spawn message
* spawn message
* spawn message
* spawn message
2020-06-13 03:47:45 -04:00
d2967d9305
update hparams, allow OmegaConf ( #2047 )
...
* DictConf
* inits
* Apply suggestions from code review
Co-authored-by: Omry Yadan <omry@fb.com>
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* atrib
* wip
* wip
* wip
* added hparams test
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* wip
* Update test_hparams.py
* added hparams test
* added hparams test
* pep8
* pep8
* pep8
* docs
* wip
* wip
* clean
* review @omry
* Update docs/source/hyperparameters.rst
Co-authored-by: Omry Yadan <omry@fb.com>
Co-authored-by: Omry Yadan <omry@fb.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-06-08 07:19:34 -04:00
8211256c46
data transfer model hook (+ refactor) ( #1756 )
...
* refactor and added hook
variant a
variant b
add test
revert rename
add changelog
docs
* resolve merge duplication
* overridden typo
* fix test
* tpu id
* raise if TPU not available
* re-use apply_to_collection function for parsing collections
* comment
* make utility function available to user
* documentation
* move changelog entry to top
* fix tpu transfer call
* fix call
* remove hardcoded string
* improve test
* call model hook by default
* Apply suggestions from code review
* rename utility function
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-06-02 21:45:19 -04:00
58a467dd68
model checkpint on rank_zero_only & global rank state ( #1408 )
...
* try delete in async or DDP us0-ecase
* changelog
* add model chekpoint rank
* simple delete
* flake8
* use global rank
* chnagelog
* fix review
* fix import
* proposal
* proposal
* proposal
* improve proposal (fix problems with method call self)
* cleaning
Co-authored-by: Adrian Wälchli <adrian.waelchli@students.unibe.ch>
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-04-24 17:21:00 -04:00
17f58d2e11
add rank warning ( #1428 )
...
* add rank warning
* changelog
* use rank_zero_warn
* user trainer_init
* replace warnings
* fix test
* flake8
* docs
* changelog
* bug lol
2020-04-09 14:05:46 -04:00
92f8c57ff5
cutout examples
2019-08-05 09:51:47 +02:00
Jirka Borovec