Update str

2026-10-24 12:13:33 +08:00 · 2020-04-16 19:10:37 +02:00
parent 573654c2da
commit 6d3db7d283
1 changed files with 1 additions and 2 deletions
@@ -78,8 +78,7 @@ class ToolRecommendations():
                        # iterate through all the attributes of the model to find weights of neural network layers
                        for item in trained_model.keys():
                            if "weight_" in item:
-                                key = "weight_{}".format(str(counter_layer_weights))
-                                weight = trained_model[key][()]
+                                weight = trained_model["weight_" + str(counter_layer_weights)][()]
                                model_weights.append(weight)
                                counter_layer_weights += 1
                        self.loaded_model = model_from_json(model_config)