Missing "resnet50w5" linear config in eval_linear.py
zeyuanyin opened this issue · 0 comments
zeyuanyin commented
It seems that "resnet50w5" is missing from this conditional statement.
Lines 229 to 249 in 06b1b7c
if global_avg:
if arch == "resnet50":
s = 2048
elif arch == "resnet50w2":
s = 4096
elif arch == "resnet50w4":
s = 8192
elif arch == "resnet50w5": # add
s = 10240 # add
Can you please confirm it?