Capture Activations and Attention Maps using PyTorch Hooks
Hooks allow us to interrupt the dynamic computational graph created by PyTorch in the forward or backward pass.
#COMPUTER-VISION
Posts tagged with "computer-vision"
Posts tagged with "computer-vision"