Mashiro d624fa9191 [Enhance] assert image shape before forward (#300) ... * assert image shape before forward * add unit test * enhance error message * allow gray image input * fix as comment * fix unit test * fix unit test		2022-06-28 11:46:12 +08:00
..
test_base_model	[Enhance] assert image shape before forward (#300)	2022-06-28 11:46:12 +08:00
test_wrappers	[Feat] Support FSDP Training (#304)	2022-06-21 15:32:56 +08:00
test_averaged_model.py	[Feature] Add autocast wrapper (#307)	2022-06-22 19:49:20 +08:00
test_base_module.py	[Feature] Add ModuleList Sequential and ModuleDict (#299)	2022-06-13 13:51:07 +08:00