Skip to content

Commit d44ee04

Browse files
Fix kernel_layers when not set in config. (ecmwf#93)
1 parent 29942b6 commit d44ee04

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

models/src/anemoi/models/models/encoder_processor_decoder.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
import einops
1515
import torch
1616
from hydra.utils import instantiate
17-
from omegaconf import OmegaConf
1817
from torch import Tensor
1918
from torch import nn
2019
from torch.distributed.distributed_c10d import ProcessGroup
@@ -70,7 +69,7 @@ def __init__(
7069
input_dim = self.multi_step * self.num_input_channels + self.node_attributes.attr_ndims[self._graph_name_data]
7170

7271
# read config.model.layer_kernels to get the implementation for certain layers
73-
self.layer_kernels = load_layer_kernels(OmegaConf.select(model_config, "model.layer_kernels"))
72+
self.layer_kernels = load_layer_kernels(model_config.get("model.layer_kernels", {}))
7473

7574
# Encoder data -> hidden
7675
self.encoder = instantiate(

0 commit comments

Comments
 (0)