pointnet2 working - TODO: Eval!

2020-05-26 21:44:56 +02:00
parent 4b4051c045
commit 77ea043907
5 changed files with 138 additions and 73 deletions
--- a/utils/tools.py
+++ b/utils/tools.py
@ -3,6 +3,15 @@ import shelve
 from pathlib import Path


+def fix_all_random_seeds(config_obj):
+    import numpy as np
+    import torch
+    import random
+    np.random.seed(config_obj.main.seed)
+    torch.manual_seed(config_obj.main.seed)
+    random.seed(config_obj.main.seed)
+
+
 def write_to_shelve(file_path, value):
    check_path(file_path)
    file_path.parent.mkdir(exist_ok=True, parents=True)