Seg-AN Train the network by running $ python train.py Options -e n, --epochs n - run the model for n epochs -d n, --device n - set the device to n