added anomaly detection for images

InnopolisUni · Jul 24, 2024 · f967acd · f967acd
1 parent d7a0e64
commit f967acd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -13,7 +13,7 @@
 
 from pckg_util import check_gpu_and_torch_compatibility
 
-# check_gpu_and_torch_compatibility()
+check_gpu_and_torch_compatibility()
 # os.environ["WANDB_DISABLED"] = "true"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
@@ -56,4 +56,4 @@ def main(config) -> float:
         )
     sys.argv.append("hydra.run.dir=./logs")
     sys.argv.append("hydra.job.chdir=True")
-    main()
+    main()