-
Notifications
You must be signed in to change notification settings - Fork 6
/
kinetics400.sh
executable file
·43 lines (37 loc) · 1.03 KB
/
kinetics400.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/bash
if [ ! -f .env ]
then
export $(cat .env | xargs)
fi
PROJECT=models/coslow
DATASET=kinetics400
PRECISION=32
# Run test sequence ######################
python $PROJECT/main.py \
--id CoSlow_kinetics_frames_8 \
--dataset $DATASET \
--seed 42 \
--gpus 1 \
--from_hparams_file models/coslow/hparams/slow_8x8_kinetics.yaml \
--finetune_from_weights models/coslow/weights/slow_8x8_kinetics.pyth \
--co3d_forward_mode init_frame \
--batch_size 1 \
--benchmark True \
--logging_backend wandb \
--num_workers 4 \
--precision $PRECISION \
--test \
python $PROJECT/main.py \
--id CoSlow_kinetics_frames_64 \
--dataset $DATASET \
--seed 42 \
--gpus 1 \
--from_hparams_file models/coslow/hparams/slow_8x8_kinetics.yaml \
--finetune_from_weights models/coslow/weights/slow_8x8_kinetics.pyth \
--co3d_forward_mode init_frame \
--batch_size 1 \
--logging_backend wandb \
--num_workers 4 \
--precision $PRECISION \
--temporal_window_size 64 \
--test \