Update run.py

run.py

@@ -21,8 +21,8 @@ from google_solver.google_model import evaluate_google_model
 with open('params.json', 'r') as f:
     params = json.load(f)
 
-# تحديد مجلد آمن لحفظ الملفات
-safe_data_dir = os.path.join(os.getcwd(), "data")
+# تحديد مجلد آمن لحفظ الملفات داخل $HOME
+safe_data_dir = "/home/user/data"
 os.makedirs(safe_data_dir, exist_ok=True)
 
 # حفظ نسخة من الإعدادات
@@ -132,7 +132,7 @@ for epoch in range(params['num_epochs']):
         del batch
 
     # التحقق من الأداء
-    if epoch % 1 == 0:
+    if epoch % 5 == 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 % 2 == 0:
+            if epoch % 10 == 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"))