import argparse
parser=argparse.ArgumentParser()
parser.add_argument('--batch_size',type=int,default=1,help='batch size, default: 1')
parser.add_argument('--image_size',type=int,default=256,help='image size, default: 256')
parser.add_argument('--use_lsgan',type=bool,default=True,
help='use lsgan (mean squared error) or cross entropy loss, default: True')
parser.add_argument('--norm',type=str,default='instance',
help='[instance, batch] use instance norm or batch norm, default: instance')
parser.add_argument('--lambda1',type=float, default=10.0)
parser.add_argument('--lambda2',type=float, default=10.0)
parser.add_argument('--learning_rate',type=float, default=2e-4)
parser.add_argument('--beta1',type=float, default=0.5)
parser.add_argument('--pool_size',type=int,default=50)
parser.add_argument('--ngf',type=int,default=64,
help='number of gen filters in first conv layer, default: 64')
parser.add_argument('--x',type=str,default='datasets/man2woman/man.tfrecords')
parser.add_argument('--y',type=str,default='datasets/man2woman/woman.tfrecords')
parser.add_argument('--load_model',type=str,default='20170828-1941')
FLAGS, unparsed = parser.parse_known_args()
for k, v in FLAGS._get_kwargs():
print(k, "=", v)
运行之后得到:
batch_size = 1
beta1 = 0.5
image_size = 256
lambda1 = 10.0
lambda2 = 10.0
learning_rate = 0.0002
load_model = 20170828-1941
ngf = 64
norm = instance
pool_size = 50
use_lsgan = True
x = datasets/man2woman/man.tfrecords
y = datasets/man2woman/woman.tfrecords