Sfoglia il codice sorgente

update hub serving doc

will-jl944 4 anni fa
parent
commit
e371964dd8
1 ha cambiato i file con 3 aggiunte e 2 eliminazioni
  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"
         "version": "1.0.0"
       },
       },
       "predict_args": {
       "predict_args": {
-        "batch_size": 1,
-        "use_gpu": false
+        "topk": 3,
+        "warmup_iters": 30,
+        "repeats": 5
       }
       }
     }
     }
   },
   },