search

klintan / pytorch-lanenet

LaneNet implementation in PyTorch
MIT License
216 stars 60 forks source link

Determining if loading a pretrained model was successful #9

Closed rahilmodi97 closed 4 years ago

rahilmodi97 commented 4 years ago

Hello, My concern is while writing python lanenet/train.py -- dataset path to the folder --pretrained path to the pretrained model. Is there any way to determine if the pretrained model has loaded successfully or not as running it with or without the --pretrained argument it is not clear whether the model has loaded or not. Can you help me with this issue?

klintan commented 4 years ago

Hm that is good point. It's probably just a matter of adding a small logging / print statement if the models i successfully loaded when running the train.py file. Feel free to add a PR and i'll review it:)