https://github.com/catalyst-team/introspection
https://github.com/catalyst-team/introspection
Last synced: 6 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/catalyst-team/introspection
- Owner: catalyst-team
- License: apache-2.0
- Created: 2021-07-22T06:04:50.000Z (about 5 years ago)
- Default Branch: main
- Last Pushed: 2022-06-17T16:17:04.000Z (about 4 years ago)
- Last Synced: 2025-10-21T18:41:13.143Z (9 months ago)
- Language: Jupyter Notebook
- Size: 3.01 MB
- Stars: 1
- Watchers: 0
- Forks: 3
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Introspection [WIP]
### tl;dr
#### Prerequirements
```bash
pip install -r ./requirements.txt
```
#### Image
```bash
# classification
PYTHONPATH=. python ./scripts/train_image.py
# classification + metric learning
PYTHONPATH=. python ./scripts/train_image.py --use-ml
```
#### Video
```bash
# 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
- [](https://colab.research.google.com/github/catalyst-team/introspection/blob/main/notebooks/train.ipynb) [Training pipeline](./notebooks/train.ipynb)
- [](https://colab.research.google.com/github/catalyst-team/introspection/blob/main/notebooks/intospection.ipynb) [Captum visualization](./notebooks/intospection.ipynb)