Hi!我想请教一下代码复现的问题。我复现了P3MNet模型,在P和NP数据集上都进行了测试,但是结果和论文有一定差距。我按照ViTAE-Transformer-Matting这篇文章中训练设置进行训练,“We train P3M-Net variants on NVIDIA Tesla V100 GPUs with a batch size of 8 for 150 epochs, which takes about 2 days.”,但是效果有差距。论文在P、NP数据集上SAD上结果为8.73和11.2300,我1卡bs8得到的结果是9.6051和15.9386,同时也尝试了4卡bs32,结果为9.6127和13.5515。想请教一下是我训练时哪儿有问题吗?还是有用到一些trick?感谢!!
Hi!我想请教一下代码复现的问题。我复现了P3MNet模型,在P和NP数据集上都进行了测试,但是结果和论文有一定差距。我按照ViTAE-Transformer-Matting这篇文章中训练设置进行训练,“We train P3M-Net variants on NVIDIA Tesla V100 GPUs with a batch size of 8 for 150 epochs, which takes about 2 days.”,但是效果有差距。论文在P、NP数据集上SAD上结果为8.73和11.2300,我1卡bs8得到的结果是9.6051和15.9386,同时也尝试了4卡bs32,结果为9.6127和13.5515。想请教一下是我训练时哪儿有问题吗?还是有用到一些trick?感谢!!