u-2-net-portrait icon indicating copy to clipboard operation
u-2-net-portrait copied to clipboard

Regarding intuitions behind certain improvements

Open tumble-weed opened this issue 4 years ago • 2 comments

Hi, thanks for the work! i was interested in certain improvements you had made, and was wondering if you had an insight about why they help. you mention mixed precision training and channels first help. could you help me understand why these have that effect?

thanks

tumble-weed avatar Oct 31 '21 03:10 tumble-weed

same confusion,hope your generous answer,thanks!

bberzh avatar Dec 02 '21 09:12 bberzh

If your hardware has tensor cores then mixed precision and channels last training heavily influences training performance:

  • https://pytorch.org/tutorials/intermediate/memory_format_tutorial.html
  • https://pytorch.org/blog/accelerating-training-on-nvidia-gpus-with-pytorch-automatic-mixed-precision/

(they do not affect model performance).

dbpprt avatar Dec 02 '21 09:12 dbpprt