fcdl94 / MiB

Official code for Modeling the Background for Incremental Learning in Semantic Segmentation https://arxiv.org/abs/2002.00718
MIT License
167 stars 44 forks source link

Remove amp and parallelization #38

Closed francescodisalvo05 closed 3 years ago

francescodisalvo05 commented 3 years ago

Replaced amp with torch.cuda.amp and removed abn and GPU parallelization, because with Colab we're not allowed to use multiple GPUs