benchmark
4b434db1 - use eval mode for vision_maskrcnn (#130163)

Commit
1 year ago
use eval mode for vision_maskrcnn (#130163) Summary: Try to fix https://github.com/pytorch/pytorch/issues/130161 The reason that `--accuracy` works is we use eval mode. While `--training` does not work since we use training mode but TorchBench does not return targets tenors. In training mode, vision_maskrcnn requires targets tensors I fix that to always use eval mode for vision_maskrcnn training. With the fix, I start see a segfault: https://gist.github.com/shunting314/5a70df3463b2a4421b2c34aa88e78d1f I'm not sure if that's due to my local setup but I think the fix in this PR is something we need any way. We can check the dashboard after the PR is in. X-link: https://github.com/pytorch/pytorch/pull/130163 Approved by: https://github.com/jansel ghstack dependencies: #129996, #129941, #130005 Reviewed By: izaitsevfb Differential Revision: D59432162 Pulled By: shunting314 fbshipit-source-id: 3a12de045de02f2b41b8366ee338e41f9a2c0e30
Author
Parents
Loading