diff options
Diffstat (limited to 'training/util.py')
-rw-r--r-- | training/util.py | 5 |
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 | |||
5 | from PIL import Image | 5 | from PIL import Image |
6 | 6 | ||
7 | 7 | ||
8 | def freeze_params(params): | ||
9 | for param in params: | ||
10 | param.requires_grad = False | ||
11 | |||
12 | |||
13 | def save_args(basepath: Path, args, extra={}): | 8 | def 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) |