Explorar el Código

update hub serving doc

will-jl944 hace 4 años
padre
commit
e371964dd8
Se han modificado 1 ficheros con 3 adiciones y 2 borrados
  1. 3 2
      docs/hub_serving_deploy.md

+ 3 - 2
docs/hub_serving_deploy.md

@@ -93,8 +93,9 @@ hub serving start --modules mobilenetv3_small_hub
         "version": "1.0.0"
       },
       "predict_args": {
-        "batch_size": 1,
-        "use_gpu": false
+        "topk": 3,
+        "warmup_iters": 30,
+        "repeats": 5
       }
     }
   },