Optimize with these tips. - Githubissues

gil-uav / semantic-image-segmentation

Semantic segmentation of static objects in orthophotos.

MIT License

2 stars 0 forks source link

Optimize with these tips. #21

Closed vegovs closed 4 years ago

vegovs commented 4 years ago

https://towardsdatascience.com/7-tips-for-squeezing-maximum-performance-from-pytorch-ca4a40951259

vegovs commented 4 years ago

[x] Use workers in DataLoaders
[x] Pin memory
[x] Avoid CPU to GPU transfers or vice-versa
[x] Construct tensors directly on GPUs
[ ] Use DistributedDataParallel not DataParallel
[x] Use 16-bit precision
[x] Profile your code
[ ] Adopting Lightning in your code