search

optimizedlearning / mechanic

MIT License
35 stars 3 forks source link

Loading Mechanic from a checkpoint throws KeyError #4

Open AlexChipev opened 6 months ago

AlexChipev commented 6 months ago

It appears that when loading Mechanic from a checkpoint it throws KeyError. I've tried to override the load_state_dict() and state_dict() methods by adding the extra mechanic state to the state_dict, but still it isn't working. Still can't figure out why.

image

AlexChipev commented 6 months ago

Seems like the solution is to reinitialize the mechanic state