Skip to content

Fixed issue when loading a single non-nn.Module object#2487

Merged
vfdev-5 merged 5 commits intopytorch:masterfrom
vfdev-5:fix-2479-checkpoint-load-optim
Feb 20, 2022
Merged

Fixed issue when loading a single non-nn.Module object#2487
vfdev-5 merged 5 commits intopytorch:masterfrom
vfdev-5:fix-2479-checkpoint-load-optim

Conversation

@vfdev-5
Copy link
Collaborator

@vfdev-5 vfdev-5 commented Feb 20, 2022

Fixed #2479

Description:

  • Fixed issue when loading a single non-nn.Module object

Check list:

  • New tests are added (if a new feature is added)
  • New doc strings: description and/or example code are in RST format
  • Documentation is updated (if required)

@github-actions github-actions bot added the module: handlers Core Handlers module label Feb 20, 2022
@vfdev-5 vfdev-5 requested a review from sdesrozis February 20, 2022 00:24
@sdesrozis
Copy link
Contributor

A few mypy issues but it seems good to me !

@vfdev-5 vfdev-5 enabled auto-merge (squash) February 20, 2022 14:34
@vfdev-5 vfdev-5 merged commit 2263d04 into pytorch:master Feb 20, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: handlers Core Handlers module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Checkpoint loading doesn't work if object doesn't contain a torch.nn.Module

2 participants