-
Notifications
You must be signed in to change notification settings - Fork 0
/
Taskfile.yml
51 lines (42 loc) · 1.28 KB
/
Taskfile.yml
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
version: "3"
# You can also ask Task to include .env like files by using the dotenv: setting:
dotenv: ['.env', '{{.ENV}}/.env.', '{{.HOME}}/.env']
# If `--` is given in the CLI, all following parameters are added to a special .CLI_ARGS variable.
# This is useful to forward arguments to another command.
tasks:
check-config:
desc: Check Hydra configuration
cmds:
- python autoencoders/train.py --cfg all --resolve {{.CLI_ARGS}}
wandb:
desc: Login to Weights & Biases
cmds:
- wandb login $WANDB_KEY
train:
desc: Train a model
cmds:
- python autoencoders/train.py {{.CLI_ARGS}}
eval-downstream:
desc: Evaluate encoders using linear baselines
cmds:
- python autoencoders/compare.py {{.CLI_ARGS}}
sources:
- autoencoders/conf/comps.yaml
- autoencoders/compare.py
generates:
- outputs/comps.json
plot-downstream:
desc: Plot encoder performance on downstream tasks
cmds:
- python autoencoders/plot.py {{.CLI_ARGS}}
sources:
- outputs/comps.json
- autoencoders/plot.py
generates:
- outputs/encoder-auc.png
- outpuers/encoder-accuracy.png
compare:
desc: Compare encoders using linear baselines
cmds:
- task: eval-downstream
- task: plot-downstream