I mesure the speed in my device

[ChengTsang]

I mesure the speed in my device, and I retrain it on the new speed.txt file.
But a mistake appeared:
Traceback (most recent call last):
File "/workspace/fbnet-pytorch/train_cifar10.py", line 105, in
speed_f=config.speed_f)
File "/workspace/fbnet-pytorch/model.py", line 84, in init
self._speed = torch.tensor(self._speed, requires_grad=False)
ValueError: expected sequence of length 8 at dim 1 (got 9)

what should I do?

[JunrQ]

Hi，@ChengTsang
I update code in file model.py, hope to help

[ChengTsang]

It helps a lot, thanks. I have an other question that which device did you test on? I see that the numbers of time table in your speed.txt is really big, but in my computer , it is really small, about 0.0002. whether you test it on the mobile phone just like the original paper ?

[JunrQ]

I test it on cpu

[ChengTsang]

ok