is it possible to parallel in training?

baiksung / MeTAL

Official PyTorch implementation of "Meta-Learning with Task-Adaptive Loss Function for Few-Shot Learning" (ICCV2021 Oral)

62 stars 13 forks source link

is it possible to parallel in training? #3

Open ligeng0197 opened 2 years ago

ligeng0197 commented 2 years ago

I tried use [0,1] and "0,1" as gpu_id, but seems not working. Does the code support parallel?

baiksung commented 2 years ago

Hi,

In its current form of the code, I'm sorry to tell you that the multi-gpu training/inference is not supported. I will update the code with the support for the multi-gpu setting in the near future.

Best, Sungyong