See models/architectures.py to see the different model architectures available, and see utils.py to see how to run and format arguments in train.py and test.py.