Fix hyperparam

dfilan · dfilan · commit dd664f96a93b · 2023-01-11T01:44:08.000Z
diff --git a/src/reward_preprocessing/vis/reward_vis.py b/src/reward_preprocessing/vis/reward_vis.py
@@ -307,7 +307,7 @@ def param_f():
                     h=input_shape[1],
                     w=input_shape[2],
                     batch=len(feature_list),
-                    sd=3,
+                    sd=1,
                 )
 
         logging.info(f"Performing vis_traditional with transforms: {transforms}")

Original file line number	Diff line number	Diff line change
`@@ -307,7 +307,7 @@ def param_f():`
`307`	`307`	`h=input_shape[1],`
`308`	`308`	`w=input_shape[2],`
`309`	`309`	`batch=len(feature_list),`
`310`		`- sd=3,`
	`310`	`+ sd=1,`
`311`	`311`	`)`
`312`	`312`
`313`	`313`	`logging.info(f"Performing vis_traditional with transforms: {transforms}")`