liaoxingyu 62ad5e1a8b bugfix for cuda tensor to numpy ... Move cuda tensor to host memory in predictor which can be used for following process.		2021-06-17 16:05:45 +08:00
..
__init__.py	finish v0.2 ddp training	2020-07-06 16:57:43 +08:00
defaults.py	bugfix for cuda tensor to numpy	2021-06-17 16:05:45 +08:00
hooks.py	fix lr scheduler warning when amp training	2021-06-02 16:35:46 +08:00
launch.py	onnx/trt support	2020-07-29 17:43:39 +08:00
train_loop.py	remove apex dependency (#442)	2021-03-23 12:12:35 +08:00