search

Brummi / MonoRec

Official implementation of the paper: MonoRec: Semi-Supervised Dense Reconstruction in Dynamic Environments from a Single Moving Camera (CVPR 2021)
MIT License
587 stars 85 forks source link

Multi card training #23

Closed ChauChorHim closed 2 years ago

ChauChorHim commented 2 years ago

Hi! The training process takes longer than I expect. Is it possible to use multi card in one machine to train the model? I tried to add --device in the args but still only one card is working. Thank you!

Brummi commented 2 years ago

Hi, sorry for the late reply! Only the bootstrap models can be trained on multiple GPUs. Here, you have to specify it in the config files as n_gpu. Best, Felix