Merge pull request #39 from i4Ds/bugfix/revert-precision

reverts precision adds full logging
i4Ds · Jun 16, 2022 · fa5cc6a · fa5cc6a
2 parents 6f05a27 + 529c2f4
commit fa5cc6a
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 7 deletions.
diff --git a/config/ce-vae/run-fhnw-full-1.yaml b/config/ce-vae/run-fhnw-full-1.yaml
@@ -37,7 +37,7 @@ data:
     test_end_date:
       value: null
     train_val_split_ratio:
-      value: 0.95
+      value: 0.9
     train_val_split_temporal_chunk_size:
       value: 14d
 predict:

diff --git a/src/sdo/sood/algorithms/ce_vae.py b/src/sdo/sood/algorithms/ce_vae.py
@@ -486,7 +486,7 @@ def main(
                              # distributed training does not yet work because the data loader lambda cannot be pickled
                              gpus=config.devices.gpus.value,
                              profiler=profiler,
-                             precision=64,
+                             precision=32,
                              accelerator="auto",
                              default_root_dir=work_dir,
                              callbacks=callbacks)

diff --git a/src/sdo/sood/models/aes.py b/src/sdo/sood/models/aes.py
@@ -1,9 +1,7 @@
 # adjusted from https://github.com/MIC-DKFZ/mood, Credit: D. Zimmerer
-import numpy as np
 import torch
 import torch.distributions as dist
 import logging
-import sys
 
 from sdo.sood.models.nets import BasicEncoder, BasicGenerator
 
@@ -108,7 +106,7 @@ def forward(self, inpt, sample=True, no_dist=False, **kwargs):
             else:
                 return x_rec, z_dist
         except Exception as e:
-            np.set_printoptions(threshold=sys.maxsize)
+            torch.set_printoptions(profile="full")
             logger.error(f"could not forward sample {mu}", e)
             raise e
 

diff --git a/src/sdo/sood/models/nets.py b/src/sdo/sood/models/nets.py
@@ -1,8 +1,6 @@
 # adjusted from https://github.com/MIC-DKFZ/mood, Credit: D. Zimmerer
-import warnings
 
 import numpy as np
-import torch
 import torch.nn as nn