[cherry-pick] Obtaining proper base model for resnet20 (#130

kevinaer · web-flow · commit 0e5e9d333468 · 2021-03-31T15:44:08.000-04:00
diff --git a/integrations/keras/prune_resnet20.py b/integrations/keras/prune_resnet20.py
@@ -77,7 +77,7 @@ def download_model_and_recipe(root_dir: str):
         dataset="cifar_10",
         training_scheme=None,
         optim_name="base",
-        optim_category=None,
+        optim_category="none",
         optim_target=None,
         override_parent_path=model_dir,
     )

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def download_model_and_recipe(root_dir: str):`
`77`	`77`	`dataset="cifar_10",`
`78`	`78`	`training_scheme=None,`
`79`	`79`	`optim_name="base",`
`80`		`- optim_category=None,`
	`80`	`+ optim_category="none",`
`81`	`81`	`optim_target=None,`
`82`	`82`	`override_parent_path=model_dir,`
`83`	`83`	`)`