search

huster-wgm / geoseg

Geoseg - A Computer Vision Package for Automatic Building Segmentation and Outline extraction
MIT License
104 stars 30 forks source link

how to get started to train? #5

Open ChengxiHAN opened 5 years ago

ChengxiHAN commented 5 years ago

I had this problem when I start to train BR-Net.py:

geoseg-master/utils/runner.py", line 332, in self.train_log = [round(x, 3) for x in train_log] TypeError: type Tensor doesn't define round method

could you tell me how to tackle this problem?Thanks very much

huster-wgm commented 5 years ago

Which version of pytorch are you using? this repo is based on pytorch 3.0.