tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

make sure to use tensorboard to capture loss for checks

Browse Source
This commit is contained in:
Wing Lian
2025-01-08 22:30:13 -05:00
parent c45ab03487
commit f7334a1719
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/e2e/integrations/test_kd.py
View File

@@ -64,6 +64,7 @@ def min_cfg(temp_dir):
"max_steps": 5, "max_steps": 5,
"output_dir": temp_dir, "output_dir": temp_dir,
"save_safetensors": True, "save_safetensors": True,
"use_tensorboard": True,
} }