Skip to content

Instantly share code, notes, and snippets.

@fjenett
Created October 25, 2022 16:26
Show Gist options
  • Save fjenett/3e21e7862f1215943db0a0a1dafc0397 to your computer and use it in GitHub Desktop.
Save fjenett/3e21e7862f1215943db0a0a1dafc0397 to your computer and use it in GitHub Desktop.
diff --git a/model/sample.py b/model/sample.py
index a4254f0..d3ebd0c 100755
--- a/model/sample.py
+++ b/model/sample.py
@@ -23,7 +23,7 @@ else:
output_path = argv[3]
search_method = "greedy" if len(argv) < 5 else argv[4]
-meta_dataset = np.load("{}/meta_dataset.npy".format(trained_weights_path))
+meta_dataset = np.load("{}/meta_dataset.npy".format(trained_weights_path), allow_pickle=True)
input_shape = meta_dataset[0]
output_size = meta_dataset[1]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment