File tree Expand file tree Collapse file tree 1 file changed +31
-0
lines changed Expand file tree Collapse file tree 1 file changed +31
-0
lines changed Original file line number Diff line number Diff line change
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
+
You can’t perform that action at this time.
0 commit comments