UsmannK / TABOR

Implementation of TABOR: A Highly Accurate Approach to Inspecting and Restoring Trojan Backdoors in AI Systems (https://arxiv.org/pdf/1908.01763.pdf)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tensorflow.python.framework.errors_impl.InvalidArgumentError: Incompatible shapes: [9,43] vs. [32,43]

linzhijianhhh opened this issue · comments

Dear author, I encountered a problem while executing the code of snooper.py.
train_badnet.py has been trained and the environment configuration is configured as required.
Appreciate your help.

1

It may be a problem about the size of your dataset, which is divisible by 32 with a remainder of 9.

commented

tabor/snooper.py
194 for idx in trange(ceil(len(x)/32)):
use int function to replace ceil function.
194 for idx in trange(int(len(x)/32)):

So sorry, I somehow never saw this. I think it is due to your image size, are you using 32x32 images?