Export all models in one go with some extra quirks

fabio-sim / Depth-Anything-ONNX

ONNX-compatible Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data

Apache License 2.0

234 stars 23 forks source link

Export all models in one go with some extra quirks #9

Closed NevermindNilas closed 3 months ago

NevermindNilas commented 3 months ago

Hi,

This pr should allow for some more options such as exporting all models in one go, the desired precision ( fp16 / fp32, TensorRT 10 sometimes throws a fit with the exports' precision ), desired opset if wanted ( 19 is the max with torch 2.2 but the nightly builds allow for opset 20 as well ) and also allow for model optimizations through ONNXSlim.

NevermindNilas commented 3 months ago

Idk if what I did was correct, haven't really done PRs before so I just commited the required changes.