Skip to content

README.md

Latest commit

 

History

History
16 lines (15 loc) · 332 Bytes

README.md

File metadata and controls

16 lines (15 loc) · 332 Bytes

Stable Matrix Factorization

Requirements

  • Linux with Python >= 3.6
  • PyTorch >= 1.1.0
  • torchvision >= 0.3.0
  • tensorboard >= 1.14.0

Quick Start

python main.py --gpu 0

You can see more options from

python main.py -h

Result files will be saved in results/.