search

dgcnz / dl2

Code for "Effect of equivariance on training dynamics"
1 stars 0 forks source link

train_rotation fails with: RuiCnRGCNN.__init__() got an unexpected keyword argument 'alpha' #59

Closed dgcnz closed 1 month ago

dgcnz commented 1 month ago

Logs

[2024-05-14 21:18:39,401][__main__][INFO] - [rank: 0] Instantiating datamodule <src.data.wang2022_datamodule.Wang2022Datamodule>
[2024-05-14 21:18:39,500][__main__][INFO] - [rank: 0] Instantiating model <src.models.wang2022_module.Wang2022LightningModule>
[2024-05-14 21:18:39,523][src.utils.utils][ERROR] - [rank: 0] 
Traceback (most recent call last):
  File "/gpfs/home6/scur0399/development/dl2/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 92, in _call_target
    return _target_(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: RuiCnRGCNN.__init__() got an unexpected keyword argument 'alpha'

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

Traceback (most recent call last):
  File "/gpfs/home6/scur0399/development/dl2/src/utils/utils.py", line 68, in wrap
    metric_dict, object_dict = task_func(cfg=cfg)
                               ^^^^^^^^^^^^^^^^^^
  File "/gpfs/home6/scur0399/development/dl2/src/train.py", line 61, in train
    model: LightningModule = hydra.utils.instantiate(cfg.model)
                             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/gpfs/home6/scur0399/development/dl2/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 226, in instantiate
    return instantiate_node(
           ^^^^^^^^^^^^^^^^^
  File "/gpfs/home6/scur0399/development/dl2/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 342, in instantiate_node
    value = instantiate_node(
            ^^^^^^^^^^^^^^^^^
  File "/gpfs/home6/scur0399/development/dl2/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 347, in instantiate_node
    return _call_target(_target_, partial, args, kwargs, full_key)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/gpfs/home6/scur0399/development/dl2/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 97, in _call_target
    raise InstantiationException(msg) from e
hydra.errors.InstantiationException: Error in call to target 'src.models.components.gcnn.rgcnn_rui.RuiCnRGCNN':
TypeError("RuiCnRGCNN.__init__() got an unexpected keyword argument 'alpha'")
full_key: model.net
[2024-05-14 21:18:39,526][src.utils.utils][INFO] - [rank: 0] Output dir: /gpfs/home6/scur0399/development/dl2/logs/train/runs/2024-05-14_21-18-39
dgcnz commented 1 month ago

Nesta solved this