Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
1.5k
stars
175
forks
source link
convert pth to onnx will give error"Exporting the operator var_mean to ONNX opset version {version} is not supported" #69
Closed
nakairika closed 2 years ago
https://github.com/google-research/big_transfer/blob/140de6e704fd8d61f3e5ea20ffde130b7d5fd065/bit_pytorch/requirements.txt#L1 upgrade pytorch to >=1.5 will fix this