Mashiro 8015d62202 [Feature] Support using gradient checkpointing in FSDP (#1382)		2023-10-09 21:04:55 +08:00
..
base_model	[Fix] Adapt to PyTorch v2.1 on Ascend (#1332)	2023-09-01 16:55:45 +08:00
wrappers	[Feature] Support using gradient checkpointing in FSDP (#1382)	2023-10-09 21:04:55 +08:00
__init__.py	[Experimental] Add support for FSDP (#1213)	2023-06-29 15:19:33 +08:00
averaged_model.py	[Feature] Support Pure Python style Configuration File (#1071)	2023-06-16 12:52:07 +08:00
base_module.py	[Enhance] Support skipping initialization in `BaseModule` (#1263)	2023-07-25 12:56:42 +08:00
efficient_conv_bn_eval.py	[Enhance] Use graph transform to deal with more general cases for efficient_conv_bn_eval (#1259)	2023-07-26 17:52:57 +08:00
test_time_aug.py	Make TTAModel compatible with FSDP (#611)	2022-12-27 21:39:43 +08:00
utils.py	[Enhancement] Unify the use of print_log and logger (#997)	2023-03-17 12:31:51 +08:00
weight_init.py	[Docs] Resolve warnings in sphinx build (#915)	2023-02-08 20:36:57 +08:00