convert.yaml

# @package _global_

# specify here default training configuration
defaults:
  - _self_
  - model: regressor.yaml

  # enable color logging
  - override hydra/hydra_logging: colorlog
  - override hydra/job_logging: colorlog

# pretty print config at the start of the run using Rich library
print_config: True

# disable python warnings if they annoy you
ignore_warnings: True

# root
root: ${hydra:runtime.cwd}

# TODO: cahnge to your checkpoint file
checkpoint_dir: ${root}/weights/last.ckpt

# dump dir
dump_dir: ${root}/weights

# input sample shape
input_sample:
  __target__: torch.randn
  size: (1, 3, 224, 224)

# convert to
convert_to: "pytorch" # [pytorch, onnx, tensorrt]

# TODO: model name without extension
name: ${dump_dir}/pytorch-kitti

# convert_to: "onnx" # [pytorch, onnx, tensorrt]
# name: ${dump_dir}/onnx-3d-0817-5