|
import warnings
|
|
from torch.onnx import _CAFFE2_ATEN_FALLBACK
|
|
|
|
if not _CAFFE2_ATEN_FALLBACK:
|
|
warnings.warn("Caffe2 support is not fully enabled in this PyTorch build. "
|
|
"Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.")
|