pytorch/examples

train from last weight

Open

#797 opened on Jul 2, 2020

View on GitHub
 (1 comment) (0 reactions) (0 assignees)Python (21,634 stars) (9,429 forks)batch import
help wanted

Description

Can this project continue training from the last saved weight？I trained one epoch with seven hours.and now I want to train on it basis

Contributor guide

Tech stack
pythonpytorch
Domain
machine learningai
Issue type
feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
under 1 hour
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
stale
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
unclear
Prerequisites
None
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
40
Research direction
The issue asks if the project supports resuming training from a saved weight. First, identify which specific example the user is referring to, as the repository contains multiple examples. Check the repository's documentation and example scripts for checkpointing functionality (e.g., saving and loading model state dict). Look for existing issues or PRs related to this feature. Provide a clear answer on how to implement checkpointing in the relevant example, or if it is already supported, direct the user to the appropriate script and instructions.
train from last weight · pytorch/examples#797 | Good First Issue