Open manapshymyr-OB opened 2 years ago
The error reads like your weight file is corrupted; hard to tell from the error alone, though.
By now this project is outdated, there are major pytorch version upgrades, and so on.
Please read the note in the readme, though, this project is no longer maintained, developed, or in any other form active
https://github.com/mapbox/robosat/blob/cbb1c73328183afd2d6351b7bfa3f430b73103ea/README.md
Hello! @daniel-j-h Thank you for your project.
I am having an error during prediction:
sudo docker run --runtime=nvidia -it --rm -v $PWD:/data --ipc=host mapbox/robosat:latest-gpu predict --tile_size 512 --model /data/robosat/config/model-unet.toml --dataset /data/robosat/config/dataset-building.toml --checkpoint /data/robosat/checkpoint/checkpoint-00001-of-00010.pth /data/robosat/datasets /data/
I am running on my laptop, with GTX 1660TI. Do you think this is the reason for that? Furthermore, getting same error while trying with only CPU. Can you please comment on that?