dcGAN 파라미터들
if __name__ == '__main__':
parser = argparse.ArgumentParser()
# model hyper-parameters
parser.add_argument('--image_size', type=int, default=64) # input size
parser.add_argument('--z_dim', type=int, default=100) # random input image size
parser.add_argument('--g_conv_dim', type=int, default=64) # G-net conv size
parser.add_argument('--d_conv_dim', type=int, default=64) # D-net conv size
# training hyper-parameters
parser.add_argument('--num_epochs', type=int, default=100) # epoch
parser.add_argument('--batch_size', type=int, default=32) # batch size
parser.add_argument('--sample_size', type=int, default=100) # out sample 10x10
parser.add_argument('--num_workers', type=int, default=8) # thread
parser.add_argument('--lr', type=float, default=0.0002) # lr
parser.add_argument('--beta1', type=float, default=0.5) # momentum1 in Adam
parser.add_argument('--beta2', type=float, default=0.999) # momentum2 in Adam
# misc
parser.add_argument('--mode', type=str, default='train') # mode
parser.add_argument('--model_path', type=str, default='./models') # save path
parser.add_argument('--sample_path', type=str, default='./samples') # save path
parser.add_argument('--image_path', type=str, default='./normal') # data path ***************
parser.add_argument('--log_step', type=int , default=10) # log print step
parser.add_argument('--sample_step', type=int , default=500) # sample image saving step
config = parser.parse_args()
print(config)

댓글
댓글 쓰기