xuebinqin/BASNet

您好 我想请教关于预训练模型的问题

Roylo-bot opened this issue · 10 comments

Roylo-bot commented

我下载了您给出的预训练模型，并放在了所指定的文件夹内。
但是在运行test文件的时候，还是会从pytorch的网站上下载模型

xuebinqin commented
Roylo-bot commented

十分感谢 。 还想请问一下 ，trian当中的train_data应该放在哪里呀？刚开始接触不太懂

Roylo-bot commented

是直接新建一个trian_data文件夹在目录下面吗？

xuebinqin commented
Roylo-bot commented

十分感谢
在训练DUTS数据集的时候，我的显卡是6G的2060,batchsize只能设置为2，不然就会超了内存。
如果要跑完十万次epoch的话需要的时间太久了，请问可以怎样修改epoch的值呢？

Pandikk commented

十分感谢
在训练DUTS数据集的时候，我的显卡是6G的2060,batchsize只能设置为2，不然就会超了内存。
如果要跑完十万次epoch的话需要的时间太久了，请问可以怎样修改epoch的值呢？

我也有同样的困惑，请问你问题得到解决了吗

xuebinqin commented
ouguozhen commented

这个下载主要是用于初始化pre-trained ResNet-34. 在basnet_test.py中这些权重将会被DUTS-TR训练的模型参数覆盖。
…
On Sat, Feb 15, 2020 at 6:09 AM Roylo-bot @.***> wrote: 我下载了您给出的预训练模型，并放在了所指定的文件夹内。 但是在运行test文件的时候，还是会从pytorch的网站上下载模型 — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#36?email_source=notifications&email_token=ADSGORPEDFHRX2WX7XPPUXTRC7SQXA5CNFSM4KVYS3OKYY3PNVWWK3TUL52HS4DFUVEXG43VMWVGG33NNVSW45C7NFSM4INYTGPQ>, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADSGORPW67RQ2ZRRLREKII3RC7SQXANCNFSM4KVYS3OA .
-- Xuebin Qin PhD Candidate Department of Computing Science University of Alberta, Edmonton, AB, Canada Homepage:https://webdocs.cs.ualberta.ca/~xuebin/

实际上不用下载啊，模型也不需要load，只需要把网络定义好就行了

BarCodeReader commented

十分感谢
在训练DUTS数据集的时候，我的显卡是6G的2060,batchsize只能设置为2，不然就会超了内存。
如果要跑完十万次epoch的话需要的时间太久了，请问可以怎样修改epoch的值呢？

我也有同样的困惑，请问你问题得到解决了吗

都不看代码的吗，里面写的清清楚楚
epoch=100000
改了就完了啊。

xuebinqin commented