weight

ptran1203 · Jul 1, 2021 · f271209 · f271209
1 parent cde5b83
commit f271209
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/prediction.py b/prediction.py
@@ -313,7 +313,7 @@ def combine_prediction(
                         help="Path to input images")
     parser.add_argument("--output", dest="output_path", metavar="O", type=str,
                         default="/data/result/submission.json", help="Output file path")
-    parser.add_argument("--weight", dest="weight_path", type=str,
+    parser.add_argument("--weight", dest="weight_path", metadata="W", type=str,
                         default="pretrained_densenet121", help="Weight path")
     args = parser.parse_args()
 

diff --git a/weights/impact_weight_dense.h5 b/weights/impact_weight_dense.h5