logging in aim after resuming works

train_configs/debug_model.toml

@@ -17,7 +17,7 @@ log_freq = 1
 enable_color_printing = true
 enable_aim = true
 save_aim_folder = "aim"
-#aim_hash = "c6b4d8b340f74287b82ef928"
+aim_hash = "1d56ec7bed87438684a8da6b"
 #aim_experiment_name = "hello"
 
 [model]
@@ -38,7 +38,7 @@ gradient_accumulation_steps = 1
 seq_len = 2048
 warmup_steps = 2  # lr scheduler warm up, normally 20% of the train steps
 max_norm = 1.0  # grad norm clipping
-steps = 20
+steps = 30
 data_parallel_degree = -1
 tensor_parallel_degree = 1
 compile = false
@@ -49,8 +49,10 @@ pipeline_parallel_degree = 1
 enable_async_tensor_parallel = false
 
 [checkpoint]
-enable_checkpoint = false
-folder = "checkpoint"
+enable_checkpoint = true
+save_folder = "checkpoint"
+load_folder = "checkpoint"
+create_seed_checkpoint = false
 interval_type = "steps"
 interval = 5
 model_weights_only = false