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)

댓글

이 블로그의 인기 게시물

파이썬으로 Homomorphic Filtering 하기

파이썬으로 2D FFT/iFFT 하기: numpy 버전