Update run.py

run.py

@@ -132,7 +132,7 @@ for epoch in range(params['num_epochs']):
         del batch
 
     # التحقق من الأداء
-    if epoch % 5 == 0:
+    if epoch % 1 == 0:
         baseline_dataloader = DataLoader(baseline_dataset, batch_size=batch_size, collate_fn=baseline_dataset.collate)
         tot_cost = []
         for batch in baseline_dataloader:
@@ -176,7 +176,7 @@ for epoch in range(params['num_epochs']):
             torch.save(actor.state_dict(), model_path)
             torch.save(optimizer.state_dict(), optimizer_path)
 
-            if epoch % 10 == 0:
+            if epoch % 1 == 0:
                 torch.save(actor.state_dict(), os.path.join(safe_data_dir, f"model_epoch_{epoch}.pt"))
                 torch.save(optimizer.state_dict(), os.path.join(safe_data_dir, f"optimizer_epoch_{epoch}.pt"))