Another bug?

Open W2Q3Q1 opened this issue 3 years ago • 0 comments

Shall line 126 in main_linear.py

model.load_state_dict(state_dict)

in the same block with return?

For no matter whether cuda is available, we always wants the model to be initiated with ckpt["model"]

Apr 26 '22 01:04 W2Q3Q1