mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
0c491702c4
Define the `TORCH_ONNX_USE_EXPERIMENTAL_LOGIC` flag to allow for enabling the new torch.onnx logic and hiding them during migration and testing. The actual logic migration will happen after. Pull Request resolved: https://github.com/pytorch/pytorch/pull/132299 Approved by: https://github.com/titaiwangms
13 lines
444 B
Python
13 lines
444 B
Python
from __future__ import annotations
|
|
|
|
|
|
class ExportTypes:
|
|
"""Specifies how the ONNX model is stored."""
|
|
|
|
# TODO(justinchuby): Deprecate and remove this class.
|
|
|
|
PROTOBUF_FILE = "Saves model in the specified protobuf file."
|
|
ZIP_ARCHIVE = "Saves model in the specified ZIP file (uncompressed)."
|
|
COMPRESSED_ZIP_ARCHIVE = "Saves model in the specified ZIP file (compressed)."
|
|
DIRECTORY = "Saves model in the specified folder."
|