Commit 3074b260 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

fix freeze_attention when not set (forgot on occurence)

parent 0f14371f
......@@ -47,7 +47,12 @@ def t1_run_name_from_params(args):
"nonorm" if args.dont_normalize else "norm"
])
if args.freeze_attention:
try:
freeze_attention = args.freeze_attention
except AttributeError:
freeze_attention = False
if freeze_attention:
s = f"{s}_frozen"
return s
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment