Autoregressive Quantile Training with Policy evaluation

2024-02-21 18:11:38 +01:00
parent 2b22b6935e
commit f8823f7efa
6 changed files with 208 additions and 10 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@@ -196,7 +196,7 @@ class Trainer:

            if task:
                self.finish_training(task=task)
-                task.close()
+                # task.close()
        except Exception:
            if task:
                task.close()