NUIST-Videocoding / VCRNet

Code for VCRNet: Visual Compensation Restoration Network for No-Reference Image Quality Assessment
21 stars 6 forks source link

关于复现的问题 #1

Open HenryZhuonanShen24 opened 2 years ago

HenryZhuonanShen24 commented 2 years ago

亲爱的作者你好,我现在正在学习您的这篇文章,尝试复现。 我使用加载了您提供的VCRNet.pth和EfficientNetb0的预训练模型, 我使用的是您github上main.py使用的参数进行复现,lr=2e-4,batch=16,epoch=16。我使用的是tid2013数据集进行复现,但是复现没有达到您在论文中的效果,请问有什么复现技巧嘛。

Zonobia-A commented 2 years ago

您好,影响训练结果的因素有很多,包括模型超参数的设置、计算机性能等,建议您加载我们给出的模型作为预训练模型再进行训练

HenryZhuonanShen24 commented 2 years ago

作者你好,请问VCRNet.pth是您提供的预训练模型是吗,我训练的时候已经加载进来了。 计算机性能是 Ubuntu18.04 CUDA V11.3 一机四卡 显卡GeForce RTX 2080Ti CPU 32 GPU 4个 我使用其中一张卡训练,内存吃掉6个G左右。

------------------ 原始邮件 ------------------ 发件人: "NUIST-Videocoding/VCRNet" @.>; 发送时间: 2022年5月18日(星期三) 下午3:42 @.>; @.**@.>; 主题: Re: [NUIST-Videocoding/VCRNet] 关于复现的问题 (Issue #1)

您好,影响训练结果的因素有很多,包括模型超参数的设置、计算机性能等,建议您加载我们给出的模型作为预训练模型再进行训练

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

Zonobia-A commented 2 years ago

您可以加载我们提供的quality prediction的模型作为预训练模型

IRON-HUGHY commented 2 years ago

请问quality prediction的模型是指live_0.971那些吗,那如果我想在别的数据库上重新训练需要加载哪些预训练模型呢,超参是不是也需要调整呢