search

dllllb / pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision
Apache License 2.0
215 stars 46 forks source link

Not supported multiGPU option from pytorchlightning.Trainer #85

Closed mazitovs closed 1 year ago

mazitovs commented 1 year ago

Try to set Trainer(gpus=[0,1]), while using PtlsDataModule as data module, get such error:

AttributeError: Can't pickle local object 'PtlsDataModule.__init__.<locals>.train_dataloader'

ivkireev86 commented 1 year ago

Fixed in #88