File "/home/futong/jiarunliu/ContrastiveCrop/DDP_linear.py", line 276, in main_worker
load_weights(cfg.load, model, optimizer, resume=False)
File "/home/futong/jiarunliu/ContrastiveCrop/DDP_linear.py", line 107, in load_weights
for k, v in state_dict.items():
UnboundLocalError: local variable 'state_dict' referenced before assignment
And I have checked the aviaible keys in SimCLR's checkpoint:
Hi, Thanks for your amazing work.
I got the following error when i trying to run the linear classification experiment of SimCLR:
Command:
Error:
And I have checked the aviaible keys in SimCLR's checkpoint:
Maybe you should consider change the key here? https://github.com/xyupeng/ContrastiveCrop/blob/dab4100839972ecab0c864256d397867260e76a8/DDP_linear.py#L84-L85