RuntimeError: Failed to evaluate ConfigExpression

[thibaultdvx]

Hi,

I'm trying to use monai.bundle. With the following yaml config file:

imports: 
- $import torch
- $from scripts.dataloaders import get_dataloader

device: '$torch.device(''cuda:1'')'

# network
network_def:
  __target__: AttentionUnet
  spatial_dims: 3 
  in_channels: 1 
  out_channels: 1
  channels: [2,4]
  strides: [1,1]

net: '$@network_def.to(@device)'

# parameters
learning_rate: 0.0001
max_epochs: 2

loss_function:
  _target_: torch.nn.MSELoss()

optimizer: 
  _target_: torch.optim.Adam
  params: '[email protected]()'
  lr: '@learning_rate'

# data
train_dataloader: '$get_dataloader(True)'

val_dataloader: '$get_dataloader(False)'

# engines
trainer:
  __target__: SupervisedTrainer
  device: '@device'
  max_epochs: '@max_epochs'
  train_data_loader: '@train_dataloader'
  network: '@net'
  optimizer: '@optimizer'
  loss_function: '@loss_function'

# start training
train:
  '[email protected]()'

when I run python -m monai.bundle run train, I get the error:

RuntimeError: Failed to evaluate ConfigExpression: 
"$__local_refs['network_def'].to(__local_refs['device'])"

It seems to be related to the line: net: '$@network_def.to(@device)'. It seems to be linked to an error mentioned few lines above: AttributeError: 'dict' object has no attribute 'to'.

You can find below the full error message. I think it is not useful to give you my get_dataloader.py script because the error doesn't seem to be related, but please feel free to ask if you need more details. Thanks!

---

The full command line (train.yaml is the config file mentioned above, metadata.json is the classical metadata file of the bundle, that I didn't modify):

python -m monai.bundle run train --bundle_root "$BUNDLE" --meta_file "$BUNDLE/configs/metadata.json" --config_file "$BUNDLE/configs/train.yaml"

The full error message:

2024-03-29 16:41:49,964 - INFO - --- input summary of monai.bundle.scripts.run ---
2024-03-29 16:41:49,965 - INFO - > config_file: 'AutoEncoder_Oasis/configs/mwe.yaml'
2024-03-29 16:41:49,965 - INFO - > meta_file: 'AutoEncoder_Oasis/configs/metadata.json'
2024-03-29 16:41:49,965 - INFO - > run_id: 'train'
2024-03-29 16:41:49,965 - INFO - > bundle_root: 'AutoEncoder_Oasis'
2024-03-29 16:41:49,965 - INFO - ---


monai.bundle.workflows ConfigWorkflow.__init__:workflow_type: Current default value of argument `workflow_type=None` has been deprecated since version 1.2. It will be changed to `workflow_type=train` in version 1.4.
2024-03-29 16:41:49,965 - WARNING - Default logging file in AutoEncoder_Oasis/configs/logging.conf does not exist, skipping logging.
Traceback (most recent call last):
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/config_item.py", line 377, in evaluate
    return eval(value[len(self.prefix) :], globals_, locals)
  File "<string>", line 1, in <module>
AttributeError: 'dict' object has no attribute 'to'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/__main__.py", line 31, in <module>
    fire.Fire()
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/fire/core.py", line 143, in Fire
    component_trace = _Fire(component, args, parsed_flag_args, context, name)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/fire/core.py", line 477, in _Fire
    component, remaining_args = _CallAndUpdateTrace(
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/fire/core.py", line 693, in _CallAndUpdateTrace
    component = fn(*varargs, **kwargs)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/scripts.py", line 787, in run
    workflow.run()
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/workflows.py", line 310, in run
    return self._run_expr(id=self.run_id)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/workflows.py", line 344, in _run_expr
    return self.parser.get_parsed_content(id, **kwargs) if id in self.parser else None
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/config_parser.py", line 290, in get_parsed_content
    return self.ref_resolver.get_resolved_content(id=id, **kwargs)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/reference_resolver.py", line 193, in get_resolved_content
    return self._resolve_one_item(id=id, **kwargs)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/reference_resolver.py", line 163, in _resolve_one_item
    self._resolve_one_item(id=d, waiting_list=waiting_list, **kwargs)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/reference_resolver.py", line 163, in _resolve_one_item
    self._resolve_one_item(id=d, waiting_list=waiting_list, **kwargs)
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/reference_resolver.py", line 175, in _resolve_one_item
    item.evaluate(globals={f"{self._vars}": self.resolved_content}) if run_eval else item
  File "/network/lustre/iss02/aramis/users/thibault.devarax/miniconda3/envs/monai/lib/python3.8/site-packages/monai/bundle/config_item.py", line 379, in evaluate
    raise RuntimeError(f"Failed to evaluate {self}") from e
RuntimeError: Failed to evaluate ConfigExpression: 
"$__local_refs['network_def'].to(__local_refs['device'])"

[KumoLiu]

Hi @thibaultdvx,
Please try:

device: $torch.device('cuda:1')
network_def:
  __target__: AttentionUnet
  spatial_dims: 3 
  in_channels: 1 
  out_channels: 1
  channels: [2,4]
  strides: [1,1]
network: $@network_def.to(@device)

Thanks.

[thibaultdvx]

Hi @KumoLiu, thanks for helping!
Unfortunately, I still have the same error message...

[KumoLiu]

Use _target_ instead of __target.

[thibaultdvx]

Good catch, it worked! Thanks a lot!