PyTorch 5-particle Classifier Example

Example of training using on 5-particle practice sample

Training

To run the training, first setup the requirements (below) and then run

python main.py

Plotting

Plotting training and results coming in future scripts.

Requirements

• pytorch, of course
• ROOT6
• LArCV2
• pytorch interface, LArCVDataset

Also, download the training and validation sets from the open data webpage

• Training
• Validation

Note: as it stands, network learns, but overtrains. Working on setting proper meta-parameters and/or adding data-augmentation.

Also, you might need to set the GPU device ID in the shell. For example, to set to device 1,

export CUDA_VISIBLE_DEVICES=1

Sources

• main.py derives from the pytorch examples repo
• resnet_example.py is modified from the pytorch torchvision models resnet module