state = state[:, :, 0]*0.299 + state[:, :, 1]*0.587 + state[:, :, 2]*0.114 IndexError: index 2 is out of bounds for axis 2 with size 2

[aspnetcs]

when I run the following command python train.py --data=configs/data.json --vocab=configs/vocab.json --training=configs/training.json --model=configs/model.json --output=results/full/ the result is Loaded 76322 formulas from data/train.formulas.norm.txt Bucketing the dataset... Traceback (most recent call last): File "train.py", line 61, in main() File "/usr/local/lib/python2.7/dist-packages/click/core.py", line 764, in call return self.main(*args, **kwargs) File "/usr/local/lib/python2.7/dist-packages/click/core.py", line 717, in main rv = self.invoke(ctx) File "/usr/local/lib/python2.7/dist-packages/click/core.py", line 956, in invoke return ctx.invoke(self.callback, **ctx.params) File "/usr/local/lib/python2.7/dist-packages/click/core.py", line 555, in invoke return callback(*args, **kwargs) File "train.py", line 36, in main form_prepro=vocab.form_prepro) File "/home/rootx/me/im2latex-master/model/utils/data_generator.py", line 76, in init self._set_data_generator() File "/home/rootx/me/im2latex-master/model/utils/data_generator.py", line 84, in _set_data_generator self._data_generator = self.bucket(self._bucket_size) File "/home/rootx/me/im2latex-master/model/utils/data_generator.py", line 104, in bucket for idx, (img, formula, img_path, formula_id) in enumerate(self): File "/home/rootx/me/im2latex-master/model/utils/data_generator.py", line 201, in iter result, skip = self._process_instance(example) File "/home/rootx/me/im2latex-master/model/utils/data_generator.py", line 173, in _process_instance img = self._img_prepro(img) File "/home/rootx/me/im2latex-master/model/utils/image.py", line 52, in greyscale state = state[:, :, 0]*0.299 + state[:, :, 1]*0.587 + state[:, :, 2]*0.114 IndexError: index 2 is out of bounds for axis 2 with size 2 root@rootx-virtual-machine:/home/rootx/me/im2latex-master# How to fix it? Thanks a lot !

[jiancgao]

met the same error

[aspnetcs]

Run under GPU environment, do not run under CPU environment

[aspnetcs]

Run under GPU environment, do not run under CPU environment

--

At 2020-05-01 01:22:07, "Jiancong Gao" [email protected] wrote:

met the same error

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[jiancgao]

Run under GPU environment, do not run under CPU environment … -- At 2020-05-01 01:22:07, "Jiancong Gao" [email protected] wrote: met the same error — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Hi, later I found it is a problem with the input data. I re-ran the program and it worked. Also I modify your work a little bit to try adaptive attention and it worked pretty well. Brilliant code by you. Thanks!

[aspnetcs]

Would you like to send me your modified code and I will test it. Let's work together to write a paper, publish it, and achieve retrieval by EI or SCI. OK？

--

At 2020-05-11 05:35:27, "Jiancong Gao" [email protected] wrote:

Run under GPU environment, do not run under CPU environment … -- At 2020-05-01 01:22:07, "Jiancong Gao" [email protected] wrote: met the same error — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

Hi, later I found it is a problem with the input data. I re-ran the program and it worked. Also I modify your work a little bit to try adaptive attention and it worked pretty well. Brilliant code by you. Thanks!

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.