Closed ali-masoudi closed 5 years ago
Hey @ali-masoudi, thanks for your interest in the project. When I had trained, I went down to at most 11GB of memory with the 1080 Ti. This is what the current configs are tailored too.
I believe a regular 1080 only has 8GB so everything should work fine but you will need to play around with lower either the batch size, crop size, or both. As a note - the authors of PSPNet/ICNet mention that having a large batch size and crop size are important so I would expect your accuracy to go down if you lower these training parameters. I did noticed that smaller batch size hurt my performance as well but I would not be able to tell you by how much. Hope this helps.
hey guys I don't have access to Titan XP, but I have access to 1080 GPU on the same machine, is there any way I could train the model?