yjb767868009 / NSM_pytorch

pytorh for NSM

NSM_pytorch

This project reproduces the pytorch version of Siggraph 2019 Neural State Machine for Character-Scene Interactions.

The Github of tensorflow Neural State Machine for Character-Scene Interactions.

Requirement

torch==1.2.0

Usage

train model

python main.py --cache

test model

python main.py --test

Result