Skip to content

Latest commit

 

History

History
41 lines (21 loc) · 1.24 KB

README.md

File metadata and controls

41 lines (21 loc) · 1.24 KB

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

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