New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

SHEELE41 wants to merge 1 commit into msr-fiddle:pipedream from SHEELE41:resnet50_gpu2_experiment

SHEELE41 commented Nov 3, 2021

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.


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

12f0d36

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet