lorenmt / reco

The implementation of "Bootstrapping Semantic Segmentation with Regional Contrast" [ICLR 2022].
https://shikun.io/projects/regional-contrast
Other
162 stars 25 forks source link

code problem #40

Closed yjq767579182 closed 1 year ago

yjq767579182 commented 1 year ago

Hello, I want to ask the following problem, I know the solution train_l_data, train_l_label = train_l_dataset.next() AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'

lorenmt commented 1 year ago

Hello, sorry I don't understand your question. Are you doing multi-gpu training?

yjq767579182 commented 1 year ago

thanks for your reply I trained the model with a single GPU I have solved the problem

ohcccc commented 11 months ago

thanks for your reply I trained the model with a single GPU I have solved the problem

hello,I want to know how you do it