go client

Author: hhsecond

go_client/tensorflow_imagenet.go

@@ -63,8 +63,8 @@ func main() {
 
     model, _ := ioutil.ReadFile(modelPath)
     script, _ := ioutil.ReadFile(scriptPath)
-    client.Do("AI.MODELSET", "imagenet_model", "TF", "CPU", "INPUTS", "images", "OUTPUTS", "output", model)
-    client.Do("AI.SCRIPTSET", "imagenet_script", "CPU", script)
+    client.Do("AI.MODELSET", "imagenet_model", "TF", "CPU", "INPUTS", "images", "OUTPUTS", "output", "BLOB", model)
+    client.Do("AI.SCRIPTSET", "imagenet_script", "CPU", "SOURCE", script)
     client.Do("AI.TENSORSET", "image", "UINT8", rect.Max.X, rect.Max.Y, 3, "BLOB", imgbuf.Bytes())
 
     client.Do("AI.SCRIPTRUN", "imagenet_script", "pre_process_3ch", "INPUTS", "image", "OUTPUTS", "temp1")

go_client/torch_charrnn.go

@@ -34,7 +34,7 @@ func main() {
 	hidden := new(bytes.Buffer)
 	binary.Write(hidden, binary.BigEndian, make([]float32, arraylen))  // Array is zero valued by default
 
-	client.Do("AI.MODELSET", "charRnn", "TORCH", "CPU", model)
+	client.Do("AI.MODELSET", "charRnn", "TORCH", "CPU", "BLOB", model)
 	client.Do("AI.TENSORSET", "hidden", "FLOAT", n_layers, batch_size, hidden_size, "BLOB", hidden.Bytes())
 	client.Do("AI.TENSORSET", "prime", "INT64", 1, "VALUES", 6)
 	client.Do("AI.MODELRUN", "charRnn", "INPUTS", "prime", "hidden", "OUTPUTS", "out")

go_client/torch_imagenet.go

@@ -63,8 +63,8 @@ func main() {
 
     model, _ := ioutil.ReadFile(modelPath)
     script, _ := ioutil.ReadFile(scriptPath)
-    client.Do("AI.MODELSET", "imagenet_model", "TORCH", "CPU", model)
-    client.Do("AI.SCRIPTSET", "imagenet_script", "CPU", script)
+    client.Do("AI.MODELSET", "imagenet_model", "TORCH", "CPU", "BLOB", model)
+    client.Do("AI.SCRIPTSET", "imagenet_script", "CPU", "SOURCE", script)
     client.Do("AI.TENSORSET", "image", "UINT8", rect.Max.X, rect.Max.Y, 3, "BLOB", imgbuf.Bytes())
 
     client.Do("AI.SCRIPTRUN", "imagenet_script", "pre_process_3ch", "INPUTS", "image", "OUTPUTS", "temp1")