RoundtTble
/

dinov2_vitl14_trt_a4000

Model card Files Files and versions Community

RoundtTble commited on Jul 20, 2023

Commit

b6a8249

•

1 Parent(s): c02a81c

Change model to FP16

Files changed (1) hide show

Makefile +2 -1

Makefile CHANGED Viewed

@@ -19,7 +19,8 @@ trt:
 		"trtexec --onnx=onnx/model.onnx --saveEngine=$(MODEL_PATH)/model.plan \
 			--minShapes=input:1x3x560x560 \
 			--optShapes=input:2x3x560x560 \
-			--maxShapes=input:8x3x560x560"
 triton:

 		"trtexec --onnx=onnx/model.onnx --saveEngine=$(MODEL_PATH)/model.plan \
 			--minShapes=input:1x3x560x560 \
 			--optShapes=input:2x3x560x560 \
+			--maxShapes=input:8x3x560x560 \
+			--fp16"
 triton: