Thytu / SMIT

SMIT: A Simple Modality Integration Tool
MIT License
15 stars 3 forks source link

SMIT should supports multi-gpu training #10

Open Thytu opened 6 months ago

Thytu commented 6 months ago

Currently SMIT will only use either the cpu when no gpu available, or cuda:0 when at least one gpu is found. This considerably limits the size performance of SMIT when used on a multi-gpu instance.

SMIT should by default use every gpu unless specified otherwise through the cfg.