Search Unity

  1. Unity 6 Preview is now available. To find out what's new, have a look at our Unity 6 Preview blog post.
    Dismiss Notice
  2. Unity is excited to announce that we will be collaborating with TheXPlace for a summer game jam from June 13 - June 19. Learn more.
    Dismiss Notice
  3. Dismiss Notice

Question Resume training from specific checkpoint with Pytorch?

Discussion in 'ML-Agents' started by mbaske, Jan 7, 2021.

  1. mbaske


    Dec 31, 2017
    Hi - With Tensorflow, I was just editing the checkpoints textfile if I wanted to resume training from a specific checkpoint. What do I do when using Pytorch ( looks wild..)? Thanks!
  2. celion_unity


    Jun 12, 2019
    Looks like we always load from "", so you should just be able to copy (or symlink) the checkpoint you want to that name.

    We have MLA-1517 in our internal tracker to provide a better interface for this on the command line. It should be a pretty straightforward PR (I think) if anybody reading this wants to try to implement it.
    Raikir-i-sh and mbaske like this.