Update train.py - Githubissues

StrongResearch / isc-demos

Deep learning examples for the Instant Super Computer

13 stars 1 forks source link

Update train.py #57

Closed adam-peaston-SC closed 1 year ago

adam-peaston-SC commented 1 year ago

Removed preamble from original pytorch tutorial

Description

General description of the changes.

Model details

What specific models did you add or modify?

Steps to reproduce any training

What settings did you use, batch size, lr, number of nodes, etc

Training results

A gist of logs, a link to tensorboard etc if relevant.

Things done

General:

[ ] Linting (ruff, black, isort)

If adding a demo model training script:

[ ] Atomic saving - saving is done atomically to avoid corruption
[ ] Checkpointing - model can successfully save a checkpoint and resume.
[ ] Completed a full run