error，what happend? python train.py --config=config/toponet_vitb_512_cityscale.yaml

htcr / sam_road

Segment Anything Model for large-scale, vectorized road network extraction from aerial imagery. CVPRW 2024

https://arxiv.org/pdf/2403.16051.pdf

MIT License

151 stars 18 forks source link

Closed kuaiqushangzixiba closed 7 months ago

kuaiqushangzixiba commented 7 months ago

EchoQiHeng commented 3 months ago

May I ask how you solved it? Is it a problem with the torch version?

EchoQiHeng commented 3 months ago

May I ask how you solved it? Is it a problem with the torch version?

Solved. pairs[:, :, 0].long()