KentoNishi/torch-mutable-modules
Use in-place and assignment operations on PyTorch module parameters with support for autograd.
Stars: 7Language: Python
Give AlbumentationsX a star on GitHub — it powers this leaderboard
Star on GitHubUse in-place and assignment operations on PyTorch module parameters with support for autograd.