Aufgrund einer Störung des s3 Storage, könnten in nächster Zeit folgende GitLab Funktionen nicht zur Verfügung stehen: LFS, Container Registry, Job Artifacs, Uploads (Wiki, Bilder, Projekt-Exporte). Wir bitten um Verständnis. Es wird mit Hochdruck an der Behebung des Problems gearbeitet. Weitere Informationen zur Störung des Object Storage finden Sie hier: https://maintenance.itc.rwth-aachen.de/ticket/status/messages/59-object-storage-pilot

Commit b70389d7 authored by Thomas Michael Timmermanns's avatar Thomas Michael Timmermanns
Browse files

Fixed test error

parent d760eed4
Pipeline #56341 passed with stage
in 1 minute and 4 seconds
......@@ -13,8 +13,8 @@ class CNNPredictor_cifar10_cifar10Classifier_net{
public:
const std::string json_file = "model/cifar10_cifar10Classifier_net/net_newest-symbol.json";
const std::string param_file = "model/cifar10_cifar10Classifier_net/net_newest-0000.params";
const std::vector<std::string> input_keys = {"data"};
//const std::vector<std::string> input_keys = {"data"};
const std::vector<std::string> input_keys = {"data"};
const std::vector<std::vector<mx_uint>> input_shapes = {{1,3,32,32}};
const bool use_gpu = false;
......@@ -28,8 +28,8 @@ public:
if(handle) MXPredFree(handle);
}
void predict(const vector<float> &data,
vector<float> &softmax){
void predict(const std::vector<float> &data,
std::vector<float> &softmax){
MXPredSetInput(handle, "data", data.data(), data.size());
//MXPredSetInput(handle, "data", data.data(), data.size());
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment