Tensor size problem with increasing the batch effect

theislab / multimil

Multimodal weakly supervised learning to identify disease-specific changes in single-cell atlases

https://multimil.rtfd.io/

BSD 3-Clause "New" or "Revised" License

18 stars 3 forks source link

Closed M0hammadL closed 2 years ago

M0hammadL commented 2 years ago

As discussed when we increase the batch size >256 the model training fails