singularity shell -e docker://atlasml/ml-base:latest
.
├── __init__.py #
├── nn_inputs.py # Get pytorch tensors (inputs) from ROOT file
├── nn_models.py # Simple deep neural network model
├── nn_utils.py # Helper functions
├── README.md # This file
├── train.py # Training and testing scripts
└── convert.py # pytorch -> onnx model conversion
mkdir output
python train.py
python convert.py
/scratchfs/atlas/bowenzhang/public/ntuple