What is the purpose of this line of code?

Source: Deep Learning on Medium

I searched the PyTorch documentation, and it states that the torch.nn.Module.register_buffer attribute “adds a persistent buffer to the…

Continue reading on Medium »