dddd_trainer icon indicating copy to clipboard operation
dddd_trainer copied to clipboard

Published 20 hours ago • verified publisher icon sml2h3

nets文件夹下的__init__.py的“export_onnx”函数多出了一个参数

Open YEUNGWAIPANG opened this issue 3 years ago • 3 comments

捕获 屏幕截图 2022-04-30 025144 如上两图，如果多出了“_retain_param_name=False”参数在训练完成导出模型时会报错，将其去掉后就成功了。 屏幕截图 2022-04-30 025540

YEUNGWAIPANG avatar Apr 29 '22 18:04 YEUNGWAIPANG

请问下这个模型要训练多久呀，100张图片我训练了一晚上还没好

JackMa777 avatar Jan 20 '23 04:01 JackMa777

请问下这个模型要训练多久呀，100张图片我训练了一晚上还没好

半个小时啊 最后一张图有写 你的电脑有显卡吗？如果有的话建议用“torch.cuda.available()”检查一下你的pytorch能不能用到GPU。我的显卡不算好也没这么夸张 不过我现在是用pytorch自己写模型训练了哈哈哈

YEUNGWAIPANG avatar Jan 20 '23 04:01 YEUNGWAIPANG

torch.cuda.available()

image 显卡是1660Ti，pytorch可以用，再训练一上午看看，自己写模型厉害了

JackMa777 avatar Jan 20 '23 04:01 JackMa777