how to run main.py
mintisan opened this issue · 2 comments
mintisan commented
How to preprocess data, set the data directory, and training commands and testing commands.
muslehal commented
First you need to download the dataset
From here
https://drive.google.com/drive/mobile/folders/1ZOYpTUa82_jCcxIdTmyr0LXQfvaM9vIy
After that setup path to your dataset in this file datautils.py
the train and the test you can do it with this line in main file
parser.add_argument('--is_train', type=int, default=1, help='training the model')
muslehal commented
I forgot tell you
1 mean train model
0 mean test model
parser.add_argument('--is_train', type=int, default=1, help='training the model')