---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
Cell In[5], line 1
----> 1 model = PixelDiffusionConditional(
2 train_dataset=train_ds,
3 valid_dataset=val_ds,
4 generated_channels=ds_config.n_generated_channels,
5 condition_channels=ds_config.n_condition_channels,
6 batch_size=64,
7 cylindrical_padding=True,
8 lr=1e-4,
9 num_workers=4,
10 loss_fn=loss_fn,
11 lr_scheduler_name="Constant"
12 )
File /gpfs0/home/wider/Projects/diffusion-models-for-weather-prediction/dm_zoo/dff/PixelDiffusion.py:47, in PixelDiffusionConditional.__init__(self, generated_channels, condition_channels, train_dataset, valid_dataset, test_dataset, batch_size, lr, num_diffusion_steps_prediction, cylindrical_padding, loss_fn, num_workers, lr_scheduler_name, unet_type, schedule, num_timesteps, sampler, dims_mults)
44 self.num_workers=num_workers
45 self.lr_scheduler_name = lr_scheduler_name
---> 47 self.save_hyperparameters()
50 self.model = DenoisingDiffusionConditionalProcess(
51 generated_channels=self.generated_channels,
52 condition_channels=self.condition_channels,
(...)
60
61 )
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/core/mixins/hparams_mixin.py:111, in HyperparametersMixin.save_hyperparameters(self, ignore, frame, logger, *args)
109 if current_frame:
110 frame = current_frame.f_back
--> 111 save_hyperparameters(self, *args, ignore=ignore, frame=frame)
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:163, in save_hyperparameters(obj, ignore, frame, *args)
159 init_args = {}
161 from pytorch_lightning.core.mixins import HyperparametersMixin
--> 163 for local_args in collect_init_args(frame, [], classes=(HyperparametersMixin,)):
164 init_args.update(local_args)
166 if ignore is None:
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:135, in collect_init_args(frame, path_args, inside, classes)
133 return collect_init_args(frame.f_back, path_args, inside=True, classes=classes)
134 if not inside:
--> 135 return collect_init_args(frame.f_back, path_args, inside=False, classes=classes)
136 return path_args
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:135, in collect_init_args(frame, path_args, inside, classes)
133 return collect_init_args(frame.f_back, path_args, inside=True, classes=classes)
134 if not inside:
--> 135 return collect_init_args(frame.f_back, path_args, inside=False, classes=classes)
136 return path_args
[... skipping similar frames: collect_init_args at line 135 (5 times)]
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:135, in collect_init_args(frame, path_args, inside, classes)
133 return collect_init_args(frame.f_back, path_args, inside=True, classes=classes)
134 if not inside:
--> 135 return collect_init_args(frame.f_back, path_args, inside=False, classes=classes)
136 return path_args
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:129, in collect_init_args(frame, path_args, inside, classes)
126 if not isinstance(frame.f_back, types.FrameType):
127 return path_args
--> 129 local_self, local_args = _get_init_args(frame)
130 if "__class__" in local_vars and (not classes or isinstance(local_self, classes)):
131 # recursive update
132 path_args.append(local_args)
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:96, in _get_init_args(frame)
94 exclude_argnames = (*filtered_vars, "__class__", "frame", "frame_args")
95 # only collect variables that appear in the signature
---> 96 local_args = {k: local_vars[k] for k in init_parameters}
97 # kwargs_var might be None => raised an error by mypy
98 if kwargs_var:
File ~/.conda/envs/TORCH311/lib/python3.11/site-packages/pytorch_lightning/utilities/parsing.py:96, in <dictcomp>(.0)
94 exclude_argnames = (*filtered_vars, "__class__", "frame", "frame_args")
95 # only collect variables that appear in the signature
---> 96 local_args = {k: local_vars[k] for k in init_parameters}
97 # kwargs_var might be None => raised an error by mypy
98 if kwargs_var:
KeyError: 'ipython_dir'