search

vikasverma1077 / manifold_mixup

Code for reproducing Manifold Mixup results (ICML 2019)
483 stars 65 forks source link

Dataparallel issue #4

Open mpicci opened 5 years ago

mpicci commented 5 years ago

Hi, in manifold_mixup_hidden_ssl.py row 215. Using two GPU with DataParallel the dim of lam is two, .item() cannot convert to one single scalar.

vikasverma1077 commented 5 years ago

@mpicci Hi there! For SSL, please check our new detailed work here https://github.com/vikasverma1077/ICT

It uses single GPU for sake of simplicity.

rshaojimmy commented 5 years ago

Yeah, I have the same problem as @mpicci. I have noticed that the implementation has commented the dataparallel out.