-
Notifications
You must be signed in to change notification settings - Fork 6
/
charades.sh
executable file
·61 lines (53 loc) · 1.49 KB
/
charades.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/bin/bash
if [ ! -f .env ]
then
export $(cat .env | xargs)
fi
PROJECT=models/coslow
DATASET=charades
GPUS=2
PRECISION=32
# Run test sequence ######################
python $PROJECT/main.py \
--id CoSlow_charades_clip \
--dataset $DATASET \
--seed 42 \
--gpus $GPUS \
--from_hparams_file models/coslow/hparams/slow_8x8_charades.yaml \
--batch_size 32 \
--benchmark True \
--logging_backend wandb \
--num_workers 4 \
--precision $PRECISION \
--co3d_forward_mode clip \
--finetune_from_weights models/coslow/weights/slow_8x8_charades.pyth \
--test \
python $PROJECT/main.py \
--id CoSlow_charades_frames_8 \
--dataset $DATASET \
--seed 42 \
--gpus $GPUS \
--from_hparams_file models/coslow/hparams/slow_8x8_charades.yaml \
--finetune_from_weights models/coslow/weights/slow_8x8_charades.pyth \
--batch_size 1 \
--benchmark True \
--logging_backend wandb \
--num_workers 4 \
--precision $PRECISION \
--co3d_forward_mode frame \
--test \
python $PROJECT/main.py \
--id CoSlow_charades_frames_64 \
--dataset $DATASET \
--seed 42 \
--gpus $GPUS \
--from_hparams_file models/coslow/hparams/slow_8x8_charades.yaml \
--finetune_from_weights models/coslow/weights/slow_8x8_charades.pyth \
--batch_size 1 \
--benchmark True \
--logging_backend wandb \
--num_workers 4 \
--precision $PRECISION \
--temporal_window_size 64 \
--co3d_forward_mode frame \
--test \