Models and states saving/loading issue

[terrainer]

The latest update introduces a bug where models and states are saved as "{current_iter}.0.pth" and {current_iter}.0.state, the models are being looked for with the new naming scheme, but the state is still loaded with the expectation that there is no ".0", and returns an error as it cannot find "{current_iter}.state".

Tested this on a clean neosr install.

[neosr-project]

Fixed, thanks for reporting.