Jeevesh8/PyTorch-Learning-NBs

PyTorch Colab Notebooks to play around key advanced features/concepts of PyTorch with important links

PyTorch-Learning-NBs

PyTorch Colab Notebooks to play around key advanced features/concepts of PyTorch with important links

register_buffer

https://discuss.pytorch.org/t/why-model-to-device-wouldnt-put-tensors-on-a-custom-layer-to-the-same-device/17964/11