iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix CUDA memory issues

Browse Source
This commit is contained in:
D-X-Y
2021-03-30 09:02:41 +00:00
parent 5fb900bcb1
commit 9fc2c991f5
7 changed files with 72 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/trade/tsf-all.sh
View File

@@ -1,7 +1,7 @@
#!/bin/bash
#
# bash scripts/trade/tsf-all.sh 0 csi300 0
# bash scripts/trade/tsf-all.sh 0 csi300 0.1
# bash scripts/trade/tsf-all.sh 0 csi300 0_0
# bash scripts/trade/tsf-all.sh 0 csi300 0.1_0
# bash scripts/trade/tsf-all.sh 1 all
#
set -e
@@ -24,6 +24,6 @@ for channel in ${channels}
do
for depth in ${depths}
do
python exps/trading/baselines.py --alg TSF-${depth}x${channel}-d${drop} --gpu ${gpu} --market ${market}
python exps/trading/baselines.py --alg TSF-${depth}x${channel}-drop${drop} --gpu ${gpu} --market ${market}
done
done

10
scripts/trade/tsf.sh
View File

@@ -1,9 +1,9 @@
#!/bin/bash
#
# bash scripts/trade/tsf.sh 0 csi300 3 0
# bash scripts/trade/tsf.sh 0 csi300 3 0.1
# bash scripts/trade/tsf.sh 1 csi100 3
# bash scripts/trade/tsf.sh 1 all 3
# bash scripts/trade/tsf.sh 0 csi300 3 0_0
# bash scripts/trade/tsf.sh 0 csi300 3 0.1_0
# bash scripts/trade/tsf.sh 1 csi100 3 0.2_0
# bash scripts/trade/tsf.sh 1 all 3 0.1_0
#
set -e
echo script name: $0
@@ -24,6 +24,6 @@ channels="6 12 24 32 48 64"
for channel in ${channels}
do
python exps/trading/baselines.py --alg TSF-${depth}x${channel}-d${drop} --gpu ${gpu} --market ${market}
python exps/trading/baselines.py --alg TSF-${depth}x${channel}-drop${drop} --gpu ${gpu} --market ${market}
done