-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMakefile
More file actions
27 lines (22 loc) · 674 Bytes
/
Makefile
File metadata and controls
27 lines (22 loc) · 674 Bytes
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
clean-data:
rm -rf ./data
mkdir data
datagen: clean-data
cp ~/Code/dl/datasets/github_issues.csv ~/github_issues.csv # put on SSD to speed up
TF_CPP_MIN_LOG_LEVEL=3 t2t-datagen --t2t_usr_dir=./issues_problem --data_dir=./data --problem=issue_to_title
rm ~/github_issues.csv
make clean-train:
rm -rf ./train
train:
t2t-trainer \
--t2t_usr_dir=./issues_problem \
--data_dir=./data \
--problem=issue_to_title \
--model=transformer \
--hparams_set=transformer_prepend \
--hparams='batch_size=1024' \
--output_dir=./train \
--eval_steps=500
tensorboard:
sudo tensorboard --logdir ./train --host 0.0.0.0 --port 80
.PHONY: clean-data datagen train tensorboard