Changing multimodal module for time sorting on cpu

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

Changing multimodal module for time sorting on cpu #159

Closed IvanSergeevPhysics closed 3 months ago

IvanSergeevPhysics commented 3 months ago

multimodal_dataset.collate_fn now also transmits indexes by which the batch is assembled in MultiModalSortTimeSeqEncoderContainer, taking into account the time order. The inference remains the same, but changes have been added to align the new MultiModalSortTimeSeqEncoderContainer.