You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
Zhihao Lin 39ed23fae8
[Enhance] Enable `exclude_frozen_parameters` for `DeepSpeedEngine._zero3_consolidated_16bit_state_dict` (#1517)
1 week ago
..
_strategy [Enhance] Enable `exclude_frozen_parameters` for `DeepSpeedEngine._zero3_consolidated_16bit_state_dict` (#1517) 1 week ago
analysis [Fix] Fix the incorrect device of inputs in get_model_complexity_info (#1130) 11 months ago
config Fix docstring of Config (#1506) 1 month ago
dataset [Fix] ConcatDataset raises error when metainfo is np.array (#1407) 5 months ago
device [Feature] Add the support for musa device support (#1453) 3 months ago
dist [Feature] Add the support for musa device support (#1453) 3 months ago
evaluator [Fix] gpu collection during evaluation (#1208) 10 months ago
fileio [Enhancement] Replace warnings.warn with print_log (#961) 1 year ago
hooks [Feature] Add the support for musa device support (#1453) 3 months ago
hub [Fix] Add torchvision_0.12.json (#820) 1 year ago
infer [Enhancement] Add an option to control whether to use progress bar in BaseInference (#1135) 11 months ago
logging Fix typos and remove fullwidth unicode chars (#1488) 2 months ago
model Fix typos and remove fullwidth unicode chars (#1488) 2 months ago
optim [Feature] Add the support for musa device support (#1453) 3 months ago
registry [Enhance] Enhance error information in build function (#1088) 8 months ago
runner Perform evaluation upon training completion (#1529) 2 weeks ago
structures [Feature] Add the support for musa device support (#1453) 3 months ago
testing [Experimental] Add support for FSDP (#1213) 9 months ago
utils [Feature] Add the support for musa device support (#1453) 3 months ago
visualization [Feature] Support custom `artifact_location` in MLflowVisBackend (#1505) 1 month ago
__init__.py [Refactor] Refactor code structure (#395) 1 year ago
version.py Bump version to 0.10.3 (#1478) 2 months ago