Conversation
sdesrozis
left a comment
There was a problem hiding this comment.
I don't understand yet why lr_finder implies to save trainer but it's a detail :)
| # restore to_save and reset trainer's state | ||
| obj = torch.load(cache_filepath.as_posix()) | ||
| trainer.state = None | ||
| trainer.load_state_dict(obj["trainer"]) |
There was a problem hiding this comment.
Can it conflict with others loads ?
There was a problem hiding this comment.
Can it conflict with others loads ?
Which others loads ?
I didn't include trainer's state dict into to_save in order to keep it immutable as Mapping requires...
There was a problem hiding this comment.
lr_finder just needs to restore state of trainer, that was a stupid remark !
because when lr finder is run, engine is run too. Thus we need to come back to the initial state. |
Fixes #880
Description:
Check list: