Lucas Pasqualin

Looking into this, thanks @ultranity

Can we replicate? I believe this was solved recently in #127617

Sorry to clarify, we would need to pipe `type_check=False` to `_iterate_state_dict` to accept Namespace types, but the specific issue related to epoch I assume would be fixed in https://github.com/pytorch/pytorch/pull/127617

@ultranity , sorry I think we just missed the deadline for 2.4, but let me see if we can get this in before 2.4.1

Shouldn't this be updated to `torch==2.3.1` or the correct version which includes `_copy_state_dict` ?