Parser=argparse. ArgumentParser (description='PyTorch Diverse Colorization')
Parser. Add_argument (' dataset_key 'help=' dataset ')
Parser. Add_argument (' -g ', '- the gpu, type=int, default=0, \
Help='gpu device id)
Parser. Add_argument (' e ', '- epochs, type=int, default=15, \
Help='Number of epochs)
Parser. Add_argument (' - b ', '-- batchsize', type=int, default=32, \
Help='Batch size')
Parser. Add_argument (' z ', '- hiddensize, type=int, default=64, \
Help='Latent vector dimension')
Parser. Add_argument (' -n ', '- nthreads, type=int, default=4, \
Help='Data loader threads')
Parser. Add_argument (' - 'em,' -- epochs_mdn ', type=int, default=7, \
Help='Number of epochs for MDN')
Parser. Add_argument (' -m ', '- nmix, type=int, default=8, \
Help='Number of diverse colorization (or the output GMM components)')
Lg parser. Add_argument (' - ', '-- logstep', type=int, default=100, \
Help='Interval to the log data)
Parser. Add_argument (' -v ', '- visdom, action=' store_true ', \
Help='Visdom the visualization')
Parser. Add_argument (' -s', '-' server, type=STR, default='http://vision-gpu-4.cs.illinois.edu', \
Help='Visdom server')
Parser. Add_argument (' -p ', '- port_num, type=int, default=8097)
An error screenshot:
CodePudding user response:
Call dataset_key variables did not give input, should be in a similar call this way your py files
Python your. Py dataset_key=XXX