akirasosa / mobile-semantic-segmentation

Real-Time Semantic Segmentation in Mobile device
MIT License
715 stars 135 forks source link

How long to train the model? #43

Closed normandra closed 6 years ago

normandra commented 6 years ago

I'm planning on mimicking your architecture on the cityscapes dataset. Do you mind sharing some details on the training phase? ( Hardware / how long).

akirasosa commented 6 years ago

No problem.

It takes about 30 min to run 80 epochs with 1080 ti.

normandra commented 6 years ago

@akirasosa Thanks, I tried the model from the develop branch and ported it to android. It didn't perform that well. How much of an improvement has there been between that branch (keras/tf implementation) and your pytorch implementation?

normandra commented 6 years ago

nevermind, I just read your blog.

Thanks for the repo