add comet

ManifoldRG · Apr 17, 2021 · 7c357fb · 7c357fb
1 parent a7c793e
commit 7c357fb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/multi_train.py b/multi_train.py
@@ -3,6 +3,7 @@
 from types import SimpleNamespace
 
 import numpy as np
+from comet_ml import Experiment
 
 from main import run_experiment