AttributeError: 'CIFAR10DataModule' object has no attribute 'train_transforms'

[gokceneraslan]

When I try to run the VAE code I get this error:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_2476/1748225191.py in <module>
      4 vae = VAE()
      5 trainer = pl.Trainer(gpus=1, max_epochs=20, callbacks=[sampler])
----> 6 trainer.fit(vae, dataset)

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in fit(self, model, train_dataloaders, val_dataloaders, datamodule, ckpt_path)
    695         self.strategy.model = model
    696         self._call_and_handle_interrupt(
--> 697             self._fit_impl, model, train_dataloaders, val_dataloaders, datamodule, ckpt_path
    698         )
    699 

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in _call_and_handle_interrupt(self, trainer_fn, *args, **kwargs)
    648                 return self.strategy.launcher.launch(trainer_fn, *args, trainer=self, **kwargs)
    649             else:
--> 650                 return trainer_fn(*args, **kwargs)
    651         # TODO(awaelchli): Unify both exceptions below, where `KeyboardError` doesn't re-raise
    652         except KeyboardInterrupt as exception:

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in _fit_impl(self, model, train_dataloaders, val_dataloaders, datamodule, ckpt_path)
    733             ckpt_path, model_provided=True, model_connected=self.lightning_module is not None
    734         )
--> 735         results = self._run(model, ckpt_path=self.ckpt_path)
    736 
    737         assert self.state.stopped

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in _run(self, model, ckpt_path)
   1103         self.__setup_profiler()
   1104 
-> 1105         self._call_setup_hook()  # allow user to setup lightning_module in accelerator environment
   1106 
   1107         # check if we should delay restoring checkpoint till later

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in _call_setup_hook(self)
   1445 
   1446         if self.datamodule is not None:
-> 1447             self._call_lightning_datamodule_hook("setup", stage=fn)
   1448         self._call_callback_hooks("setup", stage=fn)
   1449         self._call_lightning_module_hook("setup", stage=fn)

~/.miniconda/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py in _call_lightning_datamodule_hook(self, hook_name, *args, **kwargs)
   1567         if callable(fn):
   1568             with self.profiler.profile(f"[LightningDataModule]{self.datamodule.__class__.__name__}.{hook_name}"):
-> 1569                 return fn(*args, **kwargs)
   1570 
   1571     def _call_callback_hooks(

~/.miniconda/lib/python3.7/site-packages/pl_bolts/datamodules/vision_datamodule.py in setup(self, stage)
     69         """
     70         if stage == "fit" or stage is None:
---> 71             train_transforms = self.default_transforms() if self.train_transforms is None else self.train_transforms
     72             val_transforms = self.default_transforms() if self.val_transforms is None else self.val_transforms
     73 

AttributeError: 'CIFAR10DataModule' object has no attribute 'train_transforms'

I am using pl '1.7.7' and pl_bolts '0.3.2post1'.

[gokceneraslan]

It worked with pip install pytorch-lightning==1.1.8 pytorch-lightning-bolts==0.3.2.post1 but then requirements.txt is wrong I think.