https://github.com/deeplearnphysics/pytorch-resnet-example
Resnet example that trains on the 5-particle practice data
https://github.com/deeplearnphysics/pytorch-resnet-example
python pytorch resnet
Last synced: 8 months ago
JSON representation
Resnet example that trains on the 5-particle practice data
- Host: GitHub
- URL: https://github.com/deeplearnphysics/pytorch-resnet-example
- Owner: DeepLearnPhysics
- Created: 2017-12-30T07:40:51.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2021-05-11T16:10:14.000Z (almost 5 years ago)
- Last Synced: 2025-04-13T00:35:15.232Z (12 months ago)
- Topics: python, pytorch, resnet
- Language: Jupyter Notebook
- Size: 118 KB
- Stars: 15
- Watchers: 3
- Forks: 18
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 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](https://github.com/DeepLearnPhysics/larcvdataset)
Also, download the training and validation sets from the [open data webpage](http://deeplearnphysics.org/DataChallenge/)
* [Training](http://www.stanford.edu/~kterao/public_data/v0.1.0/2d/classification/five_particles/practice_train_5k.root)
* [Validation](http://www.stanford.edu/~kterao/public_data/v0.1.0/2d/classification/five_particles/practice_test_5k.root)
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](https://github.com/pytorch/examples/blob/master/imagenet/main.py)
* `resnet_example.py` is modified from the pytorch torchvision models resnet module