/torch_data_tutorial

Primary LanguagePythonMIT LicenseMIT

torch_data_tutorial

Go to [Colab] and run code cells in sequence. On the colab website, click on Runtime tab, and change runtime type to GPU. Press shift+return to run each cell in order.

The code base is heavily borrowed from Social-STGCNN.