How to train / eval on only one GPU

chenjun2hao / DDRNet.pytorch

This is the unofficial code of Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes. which achieve state-of-the-art trade-off between accuracy and speed on cityscapes and camvid, without using inference acceleration and extra data

Other

161 stars 41 forks source link

How to train / eval on only one GPU #4

Closed MateKristof closed 3 years ago

MateKristof commented 3 years ago

I would highly appreciate if You were kind and tell me, how to eval and train this model on one GPU ?

I was changed GPUS: (0, 1) to GPUS: (0,) and not working.

SHMCU commented 3 years ago

Did you try this: --nproc_per_node=1 when launching training