Skip to content

Latest commit

 

History

History
22 lines (15 loc) · 473 Bytes

README.md

File metadata and controls

22 lines (15 loc) · 473 Bytes
Raw

Requirements

  • Python >= 3.9.16
  • PyTorch >= 1.12.1
  • Platforms: Ubuntu 22.04, CUDA 11.6
  • Run conda env create -f requirements.yaml to create a new conda environment.
  • Run pip install -e torchlight.

Data Preparation

Run python data_processing.py.

Training

# Train SkateFormer
python main.py --model_name skateformer --config ./config/train/assistive_furniture/SkateFormer_j.yaml

# Train STGCN
python main.py --model_name stgcn