-
Notifications
You must be signed in to change notification settings - Fork 1
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
OK/NG classnet问题: #35
Comments
代码已经增加 |
测试 |
requirments中加入imgaug。 |
docker workspace中 运行报错:
torchvision版本太低问题。使用torch==1.6.0和torchvision==0.7.0 |
ROC曲线评价与阈值选取 这个还没测试 |
OK_NG的脚本上传,运行以下命令进行:
|
运行脚本:
运行报错:
|
这个需要配置后处理 |
需要用配置文件 |
1 similar comment
需要用配置文件 |
"post_process": { |
ai_train.py 里面有网络的名称binarynet,按classnet的配置文件重新配置一个 |
task名称也有 |
ai_train.py文件在easy_tools下,这里的参数传递与trian_task参数传递一样 |
easy_tools里面的代码很简单,你自己可以看看 |
okay,需要生成一个新的配置文件吗 |
每一个网络配置一个配置文件,配置文件尽可能的多配置 |
ROC的具体运行代码在那 |
在easyai/tools/eval_tool/evaluation_show.py |
1 similar comment
在easyai/tools/eval_tool/evaluation_show.py |
运行:
报错如下:
|
输出的结果中带有[]: |
已经修改 |
训练出来的精度都为prec1: 60.268,查看log,模型参数没有固定。 |
定位目前和模型参数,config参数无关; |
目前还是定位不到问题,目前整体流程已经走通。 |
一起看看 |
在使用Pytorch进行神经网络训练时,有时会遇到训练学习率不下降的问题。出现这种问题的可能原因有很多,包括学习率过小,数据没有进行Normalization等。不过除了这些常规的原因,还有一种难以发现的原因:在计算loss时数据维数不匹配。 |
在ce2d_loss中加入两行后训练正常:
后面主要考虑是加在dataloader中还是loss中。 |
The text was updated successfully, but these errors were encountered: