search

BBBBchan / CorrMatch

Official code for "CorrMatch: Label Propagation via Correlation Matching for Semi-Supervised Semantic Segmentation"
117 stars 8 forks source link

请问一下单机单卡可以运行吗？ #10

Closed HMeng0328 closed 5 months ago

HMeng0328 commented 5 months ago

或者说代码应该做出怎么样的修改？

BBBBchan commented 5 months ago

您好，在readme中提到，可以使用sh tools/train.sh <gpu_num> <port>运行代码。 最简单的使用单个GPU训练的方法为：sh tools/train.sh 1 <port>，供您参考。