/torch-toolkit

Set of commonly-used functions and tools for PyTorch reinforcement learning

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

torch-toolkit

Set of commonly-used functions and tools for PyTorch reinforcement learning

Installation

Since Pytorch installations can vary from system to system and are typically best done via conda, I don't include installation of pytorch/torchvision/torchaudio/cudatoolkit here.