Skip to content

Commit

Permalink
small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Mar 29, 2023
1 parent be238cf commit d0b5588
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion demos/nn-model.returnn-config.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def __call__(self, x: nn.Tensor, *, spatial_dim: nn.Dim) -> nn.Tensor:
loss.mark_as_loss("ce")

net_dict = nn.get_returnn_config().get_net_dict_raw_dict(root_module=net)
from returnn_common.utils.pprint import pprint
from returnn.util.pprint import pprint

pprint(net_dict) # just for logging purpose
return net_dict
Expand Down

0 comments on commit d0b5588

Please sign in to comment.