Closed maning closed 5 years ago
nvm, found the issue:
-v <host-folder-path>:<container-folder-path>
Inside the container use "container-folder-path"
So syntax should be:
time robosat predict --tile_size 256 --model /data/2018-07-18/model-unet.toml --dataset /data/2018-07-18/dataset-building.toml --checkpoint /data/2018-07-18/checkpoint-00040-of-00050.pth /data/2018-07-18/classify /data/2018-07-18/segmentation
Traceback (most recent call last):
File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main
"__main__", mod_spec)
File "/usr/lib/python3.6/runpy.py", line 85, in _run_code
exec(code, run_globals)
File "/usr/src/app/robosat/tools/__main__.py", line 58, in <module>
args.func(args)
File "/usr/src/app/robosat/tools/predict.py", line 68, in main
net.load_state_dict(chkpt["state_dict"])
KeyError: 'state_dict'
real 0m4.167s
user 0m0.039s
sys 0m0.030s
Error above is likely due to old checkpoints used. Closing, hopefully noobs see this and won't be confused again. ;)
@daniel-j-h Excited for the new release! I'm trying my old data using I get an error
FileNotFoundError: [Errno 2]
. Note I'm not expectingpredict
to completely finish since some of the depedencies have changed since I generated this checkpoint, but I expected to get the error beyond FileNotFoundError.Probably a docker noob mistake, any pointers?
Steps are the following:
robosat
predict