easonabc-public/paia-hw3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add cuda support

Browse Source
This commit is contained in:
Eason
2024-04-25 17:32:04 +08:00
parent 1a777943ff
commit d3ce103d67
5 changed files with 24 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
justfile
View File

@ -2,6 +2,8 @@ test:
python -m mlgame -f 30 -i ./ml/ml_play_manual_1P.py -i ./ml/ml_play_manual_2P.py . --level 8 --game_times 3
build:
cd pyr && cargo build --release
build-cuda:
cd pyr && cargo build --release --feature cuda
train level:
run level:
python -m mlgame -f 400 -i ./ml/ml_play_pyr_test.py -i ./ml/ml_play_pyr_test.py . --sound off --level {{level}} --game_times 3