Is this always true?  Can't someone call net_state_dict(ignore_status=True) and cause this part to break?  Maybe I am wrong.