2trains

Requirements

To install the requirements

pip install -r requirements.txt

Running the training scripts

To run the torch script:

python 2trains/torch_training.py

To run the keras script:

python 2trains/tf_training.py