make_parallel for network containers (#985)

* make_parallel for network containers * Address comments * Address further comments
HorizonRobotics · Aug 26, 2021 · b8cb37a · b8cb37a
1 parent cadf72f
commit b8cb37a
Show file tree

Hide file tree

Showing 13 changed files with 528 additions and 68 deletions.
diff --git a/alf/algorithms/ppg/disjoint_policy_value_network_test.py b/alf/algorithms/ppg/disjoint_policy_value_network_test.py
@@ -61,7 +61,7 @@ def test_architecture(self, is_sharing_encoder):
                 EncodingNetwork,
                 conv_layer_params=self._conv_layer_params,
                 fc_layer_params=self._fc_layer_params,
-                preprocessing_combiner=NestConcat(dim=1)),
+                preprocessing_combiner=NestConcat(dim=0)),
             is_sharing_encoder=is_sharing_encoder)
 
         # Verify that the output specs are correct