meetps / pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch
https://meetshah.dev/semantic-segmentation/deep-learning/pytorch/visdom/2017/06/01/semantic-segmentation-over-the-years.html
MIT License
3.39k stars 797 forks source link

train.py index error #177

Open arthurzhang817 opened 5 years ago

arthurzhang817 commented 5 years ago

In train.py, the output starts from "Iter [2/1000]...". I went into the code, and I found out that i starts from 0 (assuming no "resume training" scheme).

In the first iteration, i is increased to 1 immediately, but in later printing lines, they all use (i+1), which means there will never be a "Iter [1/1000]...".

The solution is to set start_iter as -1.