Open HW-VMCL opened 1 year ago
Hi, I trained the BMNet on RTX 3090 with ISTD dataset, but PSNR on valid set converge to 28.699. I can only train the net with batch_size=2 not 4 as it reported in the paper. Moreover, Could you please provide a pretrained model? Thanks!
Hi, I trained the BMNet on RTX 3090 with ISTD dataset, but PSNR on valid set converge to 28.699. I can only train the net with batch_size=2 not 4 as it reported in the paper. Moreover, Could you please provide a pretrained model? Thanks!