search

kbressem / faimed3d

Extension to fastai for volumetric medical data
MIT License
32 stars 14 forks source link

learn.unfreeze() does not work if opt_state is saved #31

Closed kbressem closed 3 years ago

kbressem commented 3 years ago

Describe the bug If a 3D CNN model is saved with an optimizer state, unfreezing the model will fail.

Expected behavior learn.unfreeze() should work for all models, regardless if the optimizer state was saved.

kbressem commented 3 years ago

Disappeared when models were restructured.