Update manager.py

manager.py

@@ -12,6 +12,18 @@ LR, CLIP_EPS, VF_COEFF, ENT_COEFF = 3e-4, 0.2, 0.5, 0.01
 CKPT_PATH = "ckpt.zst"
 CKPT_FREQ = 100
 LOG_DIR = "logs"
+
+# Try main logs dir
+try:
+    os.makedirs(LOG_DIR, exist_ok=True)
+    with open(os.path.join(LOG_DIR, "test_write"), "w") as f:
+        f.write("ok")
+    os.remove(os.path.join(LOG_DIR, "test_write"))
+except Exception:
+    LOG_DIR = "/tmp/logs"
+    os.makedirs(LOG_DIR, exist_ok=True)
+
+writer = SummaryWriter(LOG_DIR)
 DEVICE = "cpu"
 MANAGER_HOST = "0.0.0.0"
 MANAGER_PORT = 7860