-
Notifications
You must be signed in to change notification settings - Fork 28
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
代码相关问题 #2
Comments
@tianqiwanlaiqiu12 Could you please post the errors here so that I can debug for you. For thr data_shape, it is used for the attention module in OLTR, since there is a spatial flatten operation in the module. We can set the value to 4 if using cifar dataset and set the value to 7 if using ImageNet. |
非常感谢回复,我用的是mnist数据,data_shape=4,在运行netbuilder.py脚本的以下代码时报错 |
我在调试的时候当data_shape=1时是没有报错的,但是不知道这个参数是做什么的,可不可以设置为1? |
@tianqiwanlaiqiu12 很抱歉恢复的比较迟。 我更新了测试样本,请查看。支持三种数据集:mnist(backbone LeNet), cifar, ImageNet. 三种数据集经过backbone网络后产生的feature的solution分别是1x1, 4x4, 和7x7。这就是data_shape的作用,因为OLTR的attention module是将特征拉平的。希望能够解决你的困惑。 |
非常感谢您的耐心解答,给我提供了很大的帮助 |
您好,最近在研究oltr方法,发现您的代码当中netbuilder.py这个脚本当设置attmodule=True时运行报错,是什么原因呢?脚本里边data_shape设置是做什么的?期盼您的回复,万分感谢
The text was updated successfully, but these errors were encountered: