first commit

utils/log.py

@@ -0,0 +1,28 @@
+import json
+import os
+import torch
+import pandas as pd
+import numpy as np
+
+
+def save_csv_log(opt, head, value, is_create=False, file_name='results'):
+    if len(value.shape) < 2:
+        value = np.expand_dims(value, axis=0)
+    df = pd.DataFrame(value)
+    file_path = opt.ckpt + '/{}.csv'.format(file_name)
+    print(file_path)
+    if not os.path.exists(file_path) or is_create:
+        df.to_csv(file_path, header=head, index=False)
+    else:
+        with open(file_path, 'a') as f:
+            df.to_csv(f, header=False, index=False)
+            
+            
+def save_ckpt(state, opt=None, file_name = 'model.pt'):
+    file_path = os.path.join(opt.ckpt, file_name)
+    torch.save(state, file_path)
+        
+
+def save_options(opt):
+    with open(opt.ckpt + '/options.json', 'w') as f:
+        f.write(json.dumps(vars(opt), sort_keys=False, indent=4))