-
Notifications
You must be signed in to change notification settings - Fork 3
/
runtraining.sh
executable file
·44 lines (32 loc) · 1007 Bytes
/
runtraining.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/bin/sh
PROJECT_ROOT=~/Projects/FootballPred
# saving model weights etc.
SAVE_PATH=$PROJECT_ROOT/saved/$(date +%d.%m_%H:%M)
PATH_SAVED_LOGS=$SAVE_PATH/logs
PATH_SAVED_MODELS=$SAVE_PATH/models
PATH_SAVED_STATS=$SAVE_PATH/stats
mkdir $SAVE_PATH
cd $SAVE_PATH
mkdir logs
mkdir models
mkdir stats
cd $PROJECT_ROOT
PATH_LOGFILE=$PATH_SAVED_LOGS/train_log.txt
# system config
PATH_DB=$PROJECT_ROOT/data/database.sqlite
DEVICE="cpu"
BIG_DATASET=1
LOGLEVEL="INFO"
EXPERIMENT=5
# hyperparameters for training
EPOCHS=50
BATCH_SIZE=8
OPTIMIZER="Adam"
LEARNING_RATE=0.001
WEIGHT_DECAY=0.0001
#OPTIMIZER="SGD"
#LEARNING_RATE=0.01
MOMENTUM=0.01
# run
EXEC=$PROJECT_ROOT/src/main.py
python3 $EXEC --database $PATH_DB --log $PATH_LOGFILE --epochs $EPOCHS --batch_size $BATCH_SIZE --model_save_path $PATH_SAVED_MODELS --stats_save_path $PATH_SAVED_STATS --optimizer $OPTIMIZER --learning_rate $LEARNING_RATE --weight_decay $WEIGHT_DECAY --momentum $MOMENTUM --big_dataset $BIG_DATASET --experiment $EXPERIMENT