Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
run_unit_tests.sh 149 Bytes
Newer Older
Gustaf Ahdritz's avatar
Add pTM config options, spruce up pretrained script, add test script  
Gustaf Ahdritz committed
1
2 
#!/bin/bash
Gustaf Ahdritz's avatar
Add low-memory attention (still needs to be incorporated)  
Gustaf Ahdritz committed
3 
CUDA_VISIBLE_DEVICES="0"
Gustaf Ahdritz's avatar
Bring tests up to speed  
Gustaf Ahdritz committed
4
Gustaf Ahdritz's avatar
Fix another FP16 overflow, finish tests, update util scripts  
Gustaf Ahdritz committed
5 
python3 -m unittest "$@" || \
Gustaf Ahdritz's avatar
Add pTM config options, spruce up pretrained script, add test script  
Gustaf Ahdritz committed
6 
echo -e "\nTest(s) failed. Make sure you've installed all Python dependencies."