Search Unity

Question Resume training from .onnx file (ML-Agents)

Discussion in 'ML-Agents' started by Iro756, Jan 5, 2024.

  1. Iro756

    Iro756

    Joined:
    Jan 5, 2024
    Posts:
    1
    I was wondering if there's a way to resume training on a nn model instead of a run id, as the actual version of the nn is no longer working as intended. I've tried to make a new folder with a new run id and pasting the nn model inside just to try and I got the following error:
    FileNotFoundError: [Errno 2] No such file or directory: 'results\\ImitC\\Drive\\checkpoint.pt'
    (Where ImitC is the new folder and Drive is the behavior name)

    If I were to find the corresponding .pt file to this nn, would I be able to resume training from there? If so, how?