hwtwj

the command line is: DATA_ROOT=./datasets/mpi_correction_corrnormamp2depth_albedoaug_zpass niter=50 save_epoch_freq=200 mat=1 input_nc=5 output_nc=1 \ model=pix2pix align_data=1 name=my_resnet_lite_my_imageGAN_128 which_model_netG=my_resnet_lite which_model_netD=my_imageGAN_128 \ use_GAN=1 lambda_A=10 norm=none \ noise=1 lr=0.00005 resize_or_crop=resize_and_crop_ratio_rand mask_nan=1 tv_strength=1e-4 tv_weight_scheme=1 tv_weight_scale=1 \ manualSeed=1...

Hi, could you tell me your model use which train datasets, and which parameters you set when you train your model? I find the results when I test the model...

hi，I have found the cause of the issue of training speed slowly, but also thank you very much

it is GPU's memory not enough，please modify the s_hashNumBuckets values or s_hashNumSDFBlocks values in file of zParametersDefault.txt， smaller will be ok.