-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_plot_static_gpu.sh
49 lines (40 loc) · 1.22 KB
/
run_plot_static_gpu.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
#!/usr/bin/env bash
rm movie_exp/*.json movie_exp/*.jsonl
python3 dino.py \
--output_dir movie_exp \
--task_file movie_task_specs/plot-location.json \
--num_entries_per_label 20 \
--min_num_tokens 1 \
--max_output_length 5 \
--top_k 30 \
--batch_size 10
python3 scripts/movie_plot/post_process.py --task location
python3 dino.py \
--output_dir movie_exp \
--task_file movie_task_specs/plot-genre.json \
--num_entries_per_label 20 \
--min_num_tokens 1 \
--max_output_length 5 \
--top_k 30 \
--batch_size 10
python3 scripts/movie_plot/post_process.py --task genre
python3 dino.py \
--output_dir movie_exp \
--task_file movie_task_specs/plot-cast.json \
--input_file movie_exp/plot-location-dataset-clean.jsonl \
--input_file_type jsonl \
--num_entries_per_input_and_label 10 \
--min_num_tokens 1 \
--max_output_length 5 \
--top_k 30
python3 scripts/movie_plot/post_process.py --task cast
python3 dino.py \
--output_dir movie_exp \
--task_file movie_task_specs/plot-theme.json \
--input_file movie_exp/plot-genre-dataset-clean.jsonl \
--input_file_type jsonl \
--num_entries_per_input_and_label 10 \
--min_num_tokens 5 \
--max_output_length 25 \
--top_k 30
python3 scripts/movie_plot/post_process.py --task theme