TorchEngineConfig
-
說明：
TorchEngine的設定（dataclass），目前提供dtype與cuda_sync。 -
依賴
- 需要
torch（請先安裝capybara-docsaid[torchscript]）。
- 需要
-
範例
from capybara.torchengine import TorchEngineConfig
cfg = TorchEngineConfig(dtype="fp16", cuda_sync=True)
說明：TorchEngine 的設定（dataclass），目前提供 dtype 與 cuda_sync。
依賴
torch（請先安裝 capybara-docsaid[torchscript]）。範例
from capybara.torchengine import TorchEngineConfig
cfg = TorchEngineConfig(dtype="fp16", cuda_sync=True)