Skip to content

Handle the "spatial" attribute in onnx BatchNormalization op#9492

Closed
JerryShih wants to merge 1 commit intopytorch:masterfrom
JerryShih:handle-onnx-BatchNormalization-spatial-attrib
Closed

Handle the "spatial" attribute in onnx BatchNormalization op#9492
JerryShih wants to merge 1 commit intopytorch:masterfrom
JerryShih:handle-onnx-BatchNormalization-spatial-attrib

Conversation

@JerryShih
Copy link
Contributor

If we have this "spatial" attribute and its value equals to 1, we could just remove this attribute and convert this op to caffe2 SpatialBN.

@JerryShih
Copy link
Contributor Author

@houseroad @yinghai
Could you please check these changes?

@yinghai yinghai requested a review from houseroad July 18, 2018 19:15
@JerryShih
Copy link
Contributor Author

Any comment for this patch?
@houseroad

Copy link
Member

@houseroad houseroad left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@JerryShih
Copy link
Contributor Author

@houseroad
how to merge this pr to master?

@houseroad
Copy link
Member

There is a merge conflict, could you update the diff? @JerryShih

@JerryShih JerryShih force-pushed the handle-onnx-BatchNormalization-spatial-attrib branch from 52e74d0 to 63b3f22 Compare July 27, 2018 02:09
If we have this "spatial" attribute and its value equals to 1, we could
just remove this attribute and convert this op to caffe2 SpatialBN.
@JerryShih JerryShih force-pushed the handle-onnx-BatchNormalization-spatial-attrib branch from 63b3f22 to 9193ad1 Compare July 27, 2018 02:21
@JerryShih
Copy link
Contributor Author

@houseroad
I fix the conflict now. Thank you.

@ezyang
Copy link
Contributor

ezyang commented Jul 27, 2018

@pytorchbot retest this please

@JerryShih
Copy link
Contributor Author

@ezyang @houseroad
retry for "pr/caffe2-py2-cuda9.0-cudnn7-ubuntu16.04-test — Build failed" ?

@houseroad
Copy link
Member

yep, retrying

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

jramseyer pushed a commit to jramseyer/pytorch that referenced this pull request Jul 30, 2018
…#9492)

Summary:
If we have this "spatial" attribute and its value equals to 1, we could just remove this attribute and convert this op to caffe2 SpatialBN.
Pull Request resolved: pytorch#9492

Differential Revision: D8988165

Pulled By: houseroad

fbshipit-source-id: a9218dc9cd5fab43deb371f290f81285f5283231
goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
…#9492)

Summary:
If we have this "spatial" attribute and its value equals to 1, we could just remove this attribute and convert this op to caffe2 SpatialBN.
Pull Request resolved: pytorch#9492

Differential Revision: D8988165

Pulled By: houseroad

fbshipit-source-id: a9218dc9cd5fab43deb371f290f81285f5283231
zoidor added a commit to zoidor/MLTests that referenced this pull request Feb 6, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants