Skip to content

Commit

Permalink
Fix merge conflict
Browse files Browse the repository at this point in the history
  • Loading branch information
HenrikMettler committed Jan 6, 2021
1 parent 6c84881 commit d7dba98
Showing 1 changed file with 18 additions and 2 deletions.
20 changes: 18 additions & 2 deletions cgp/genome.py
Original file line number Diff line number Diff line change
Expand Up @@ -712,8 +712,24 @@ def update_parameters_from_torch_class(self, torch_cls: "torch.nn.Module") -> bo

def _initialize_unknown_parameters(self) -> None:
for region_idx, region in self.iter_hidden_regions():

self._initialize_parameter_values(region_idx, region)
node_id = region[0]
node_type = self._primitives[node_id]
assert issubclass(node_type, OperatorNode)
for parameter_name_with_idx in self._get_parameter_names_with_idx_of_node(
node_type, region_idx
):
if parameter_name_with_idx not in self._parameter_names_to_values:
self._parameter_names_to_values[
parameter_name_with_idx
] = node_type.initial_value(parameter_name_with_idx)

def _get_parameter_names_with_idx_of_node(
self, node_type: Type[OperatorNode], region_idx: int
) -> List[str]:
parameter_names_with_idx: List[str] = []
for parameter_name in node_type._parameter_names:
parameter_names_with_idx.append("<" + parameter_name[1:-1] + str(region_idx) + ">")
return parameter_names_with_idx

def _initialize_parameter_values(
self, region_idx: int, region: List[int], reinitialize: bool = False
Expand Down

0 comments on commit d7dba98

Please sign in to comment.