ifgovh/Human-attention-inspired-DNN

Fix the inconsistency of ImageNet and MNIST loader

Opened this issue · 2 comments

ifgovh commented
Fix the inconsistency of ImageNet and MNIST loader
ifgovh commented

The inconsistency was caused by glimpse sensor layer size (loc_hidden), RNN hidden layer size (hidden_szie), num_classes, num_channel. I have fixed them all. loc_hidden >= 192 (patch_sizepatch_sizenum_patch*num_chaneel). hidden_size >= 320 (the output of glimpse layer).
glimpse sensor ----> glimpse ----->RNN

ifgovh commented

That is, run:
python3 main.py --dataset_name='ImageNet' --loc_hidden=192 --hidden_size=320