Fix classy_vision test on heads_fully_connected_head_test

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 278 forks source link

Closed YosuaMichael closed 2 years ago

YosuaMichael commented 2 years ago

Summary: The heads_fully_connected_head_test on classy_vision seems to failed, see: https://www.internalfb.com/intern/test/562949985032390/

And this can be reproduced in on-demand server by running the buck command provided on the above url.

The error message:

RuntimeError: linalg.norm: If dim is specified, it must be of length 1 or 2. Got [1, 2, 3]

indicate that the test has wrong value of dim input.

This diffs try to solve this issue

Differential Revision: D36702425

facebook-github-bot commented 2 years ago

This pull request was exported from Phabricator. Differential Revision: D36702425

YosuaMichael commented 2 years ago

Get segmentation fault during build, this does not seems to be related with the PR