Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Replace deprecated arg in torchvision models (#6401)
### Description For torchvision models, the parameter `pretrained` is deprecated since 0.13 and may be removed in the future. Our `TorchVisionFC` model has been using this parameter. This PR updated `TorchVisionFC` to remove calling pretrained and instead to call `weights` with appropriate defaults without any change in the functionality (backward compatible). ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [x] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [x] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. --------- Signed-off-by: Behrooz <3968947+drbeh@users.noreply.github.com> Signed-off-by: Wenqi Li <wenqil@nvidia.com> Co-authored-by: Nic Ma <nma@nvidia.com> Co-authored-by: Wenqi Li <wenqil@nvidia.com>
- Loading branch information