summaryrefslogtreecommitdiffstats
path: root/training/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'training/util.py')
-rw-r--r--training/util.py5
1 files changed, 0 insertions, 5 deletions
diff --git a/training/util.py b/training/util.py
index a0c15cd..d0f7fcd 100644
--- a/training/util.py
+++ b/training/util.py
@@ -5,11 +5,6 @@ import torch
5from PIL import Image 5from PIL import Image
6 6
7 7
8def freeze_params(params):
9 for param in params:
10 param.requires_grad = False
11
12
13def save_args(basepath: Path, args, extra={}): 8def save_args(basepath: Path, args, extra={}):
14 info = {"args": vars(args)} 9 info = {"args": vars(args)}
15 info["args"].update(extra) 10 info["args"].update(extra)