1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
[project] name = "{{PROJECT_NAME}}" [node] name = "{{PROJECT_NAME}}" [model.policy] source_framework = "pytorch" path = "models/policy.onnx" backend = "onnxruntime" device = "cpu" [model.policy.input] shape = [1, 10] layout = "nchw" dtype = "f32"