feat: print PyTorch config when running tests

2026-09-25 21:41:26 -07:00 · 2026-06-25 17:51:04 +05:30
parent 3abe88c39f
commit faf3c154aa
4 changed files with 16 additions and 2 deletions
@@ -1,14 +1,15 @@
 model = "tiny-random/mistral-3"
 model_commit = "931aa2e5c9668fc3679e56aa44972fe18597d55d"

+seed = 12345
+print_debug_information = true
+
 batch_size = 2
 max_response_length = 10
 kl_divergence_target = 0
 n_trials = 2
 n_startup_trials = 1

-seed = 12345
-
 export_strategy = "merge"
 checkpoint_action = "restart"
 trial_index = 0