Skip to content

Commit 3237ef3

Browse files
committed
added readme
1 parent e11a303 commit 3237ef3

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

README.md

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# PyTorch 5-particle Classifier Example
2+
3+
Example of training using on 5-particle practice sample
4+
5+
Requires
6+
7+
* pytorch, of course
8+
* ROOT6
9+
* LArCV2
10+
* pytorch interface, [LArCVDataset](https://github.com/DeepLearnPhysics/larcvdataset)
11+
12+
Also, download the training and validation sets from the [open data webpage](http://deeplearnphysics.org/DataChallenge/)
13+
14+
* [Training](http://www.stanford.edu/~kterao/public_data/v0.1.0/2d/classification/five_particles/practice_train_5k.root)
15+
* [Validation](http://www.stanford.edu/~kterao/public_data/v0.1.0/2d/classification/five_particles/practice_test_5k.root)
16+
17+
18+
Note: as it stands, network learns, but overtrains. Working on setting proper meta-parameters and/or adding data-augmentation.
19+
20+
Also, you might need to set the GPU device ID in the shell. For example, to set to device `1`,
21+
22+
export CUDA_VISIBLE_DEVICES=1
23+
24+
25+
### Sources
26+
27+
* `main.py` derives from the pytorch examples [repo](https://github.com/pytorch/examples/blob/master/imagenet/main.py)
28+
* `resnet_example.py` is modified from the pytorch torchvision models resnet module
29+
30+
31+

0 commit comments

Comments
 (0)