InterDigitalInc / CompressAI

A PyTorch library and evaluation platform for end-to-end compression research

Home Page:https://interdigitalinc.github.io/CompressAI/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multi-GPU support is now experimental

xuezhongcailian opened this issue · comments

Feature

Does this repository currently support multi-GPU experiments

Motivation

Additional context

Hi, Not at the moment. For reference, there is a PR on that (#223) which requires some cleaning and a more robust / generic way to gather gpu info than using WORLD_SIZE directly.