search

YuanXue1993 / SegAN

SegAN: Semantic Segmentation with Adversarial Learning
MIT License
182 stars 58 forks source link

Errors #5

Closed SMKamrulHasan closed 5 years ago

SMKamrulHasan commented 6 years ago

File "train.py", line 114, in target = target.type(torch.FloatTensor) NameError: name 'target' is not defined

vribeiro1 commented 5 years ago

You need to run it with --cuda. CUDA_VISIBLE_DEVICES=<your_gpu_id> python train.py --cuda