search

arthurdouillard / incremental_learning.pytorch

A collection of incremental learning paper implementations including PODNet (ECCV20) and Ghost (CVPR-W21).
MIT License
383 stars 60 forks source link

can use multiple GPUs while training #63

Closed ahsan856jalal closed 1 year ago

ahsan856jalal commented 2 years ago

I have a 2-GPU system , can I use two GPU so that I donot find memery issue while working with ImageNet dataset

arthurdouillard commented 1 year ago

Use the option --device 0,1 to use the first and second gpu. Only data parallelism is implemented.