Skip to content

Latest commit

 

History

History
executable file
·
42 lines (32 loc) · 1.39 KB

README.md

File metadata and controls

executable file
·
42 lines (32 loc) · 1.39 KB

Introspection [WIP]

tl;dr

Prerequirements

pip install -r ./requirements.txt

Image

# classification
PYTHONPATH=. python ./scripts/train_image.py

# classification + metric learning
PYTHONPATH=. python ./scripts/train_image.py --use-ml

Video

# go through data/scripts/readme.md
# classification
PYTHONPATH=. python ./scripts/train_video.py  # ~11GB GPU required
PYTHONPATH=. python ./scripts/train_video.py --freeze-encoder  # ~5GB GPU required

# classification + metric learning
PYTHONPATH=. python ./scripts/train_video.py --use-ml  # ~20GB GPU required
PYTHONPATH=. python ./scripts/train_video.py --freeze-encoder --use-ml  # ~15GB GPU required

Time Series [WIP]

# unpack https://drive.google.com/file/d/1MKtGfcVNiq0p7T_XGYGyTYFDYiUH96fS/view?usp=sharing to data/abide

PYTHONPATH=. python ./scripts/train_ts.py
PYTHONPATH=. python ./scripts/train_ts.py --use-ml

Demo