uber-research / UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Other
648 stars 120 forks source link

Model fails to transfer to gpu #107

Closed discretecoder closed 4 years ago

discretecoder commented 4 years ago

when I try to train on my own dataset, the model fails to generate. train_model = eval(config.symbol)().cuda() is just executed repeatedly, is there a solution?

thanks