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.
#DEEP-LEARNING
Posts tagged with "deep-learning"
Posts tagged with "deep-learning"