modify runtime/image_classification/models/resnet50/gpus=2/__init__.py - Githubissues

msr-fiddle / pipedream

MIT License

379 stars 117 forks source link

modify runtime/image_classification/models/resnet50/gpus=2/init.py #67

Open SHEELE41 opened 3 years ago

SHEELE41 commented 3 years ago

Hello, I found some bug in resnet50 gpus 2 experiment code. A keyerror was raised when running the experiment.

Traceback (most recent call last):
  File "main_with_runtime.py", line 585, in <module>
    main()
  File "main_with_runtime.py", line 146, in main
    input_tensor = torch.zeros(tuple(training_tensor_shapes[input]),
KeyError: 'input'

So I modified runtime/image_classification/models/resnet50/gpus=2/init.py file and saw it works.