Igor Kasianenko b3992f326d fix is_xpu import 2 ヶ月 前
..
__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 ヶ月 前
config_utils.py ef99b32e48 update refs 3 ヶ月 前
dataset_utils.py 0ca612fff3 fix more imports 3 ヶ月 前
flop_utils.py c0b6de8632 rename_folder 3 ヶ月 前
fsdp_utils.py b3992f326d fix is_xpu import 2 ヶ月 前
memory_utils.py c0b6de8632 rename_folder 3 ヶ月 前
plot_metrics.py c0b6de8632 rename_folder 3 ヶ月 前
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 ヶ月 前