[Error] AttributeError: 'Trainer' object has no attribute '_data_connector'

KoBart 를 사용해 모델을 전이학습 하려고 하는데, pytorch_lightning에서 계속 오류가 발생.

 

    optim_conf = model.configure_optimizers()
  File "train.py", line 101, in configure_optimizers
    data_len = self.setup_steps(self)
  File "train.py", line 84, in setup_steps
    train_loader = self.trainer._data_connector._train_dataloader_source.dataloader()
AttributeError: 'Trainer' object has no attribute '_data_connector'

 

1. 에러 원인

---

pytorch_lightning 의 버전이 안맞았다.

Kobart 코드를 뜯어보니까 setup_steps라는 함수에 관련 이슈가 링크되어 있다.

    def setup_steps(self, stage=None):
        # NOTE There is a problem that len(train_loader) does not work.
        # After updating to 1.5.2, NotImplementedError: `train_dataloader` · Discussion #10652 · PyTorchLightning/pytorch-lightning https://github.com/PyTorchLightning/pytorch-lightning/discussions/10652
        train_loader = self.trainer._data_connector._train_dataloader_source.dataloader()
        return len(train_loader)

 

 

2. 해결 방법

---

$ pip install pytorch_lightning==1.5.2

 

 

3. 참고링크

---

https://github.com/PyTorchLightning/pytorch-lightning/discussions/10652