Skip to content

Latest commit

 

History

History
 
 

01_pytorchDatasetAPI

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

PyTorch Example Data Pipeline

Led by J. Taylor Childers from ALCF

This is very similar to the TensorFlow example covered in the README one level up.

Example submission script for ThetaGPU is provided.

Submit to ThetaGPU using:

qsub -A <project> -q <queue> submit_thetagpu.sh

During this training program, you should use:

qsub -A ALCFAITP -q single-gpu submit_thetagpu.sh

All log files go into the logdir/ folder.