python train.py --epochs 100 --batch-size 1000 --lr 0.0005 --hidden 16  
    --num-atoms 20 --num-classes 5 --num_neighbors 7 --conv_type SchNet --dmin 0 
    --dmax 8. --step 0.5 --dist-data dists.dat --nbr-data nbrs.dat --residual  --train

python train.py --epochs 100 --batch-size 1000 --lr 0.0005 --hidden 16 
    --num-atoms 20 --num-classes 5 --num-neighbors 7 --conv_type SchNet --dmin 0
    --dmax 8. --step 0.5 --dist-data dists.dat --nbr-data nbrs.dat --residual --trained-model
    logs/logs_99.pt