Igor Kasianenko b3992f326d fix is_xpu import 2 månader sedan
..
__init__.py c51d3af6c9 Use auto instead of bf16 to determine model dtype; Add comments to describe the various use_fp16 and pure_bf16 options 2 månader sedan
config_utils.py ef99b32e48 update refs 3 månader sedan
dataset_utils.py 0ca612fff3 fix more imports 3 månader sedan
flop_utils.py c0b6de8632 rename_folder 3 månader sedan
fsdp_utils.py b3992f326d fix is_xpu import 2 månader sedan
memory_utils.py c0b6de8632 rename_folder 3 månader sedan
plot_metrics.py c0b6de8632 rename_folder 3 månader sedan
train_utils.py c51d3af6c9 Use auto instead of bf16 to determine model dtype; Add comments to describe the various use_fp16 and pure_bf16 options 2 månader sedan