15 lines
465 B
Bash
15 lines
465 B
Bash
#!/usr/bin/env bash
|
|
|
|
# install APEX, see https://github.com/NVIDIA/apex#linux
|
|
# to imitate SLURM set only single node
|
|
export SLURM_LOCALID=0
|
|
|
|
# use this to run tests
|
|
rm -rf _ckpt_*
|
|
rm -rf ./lightning_logs
|
|
python -m coverage run --source pytorch_lightning -m py.test pytorch_lightning tests pl_examples -v --flake8
|
|
python -m coverage report -m
|
|
|
|
# specific file
|
|
# python -m coverage run --source pytorch_lightning -m py.test -k test_trainer.py --flake8 --durations=0
|