forked from ashleve/lightning-hydra-template
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathdebug.yaml
45 lines (33 loc) · 910 Bytes
/
debug.yaml
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
# @package _global_
# to execute this experiment run:
# python train.py experiment=example
defaults:
- override /data: ptg
- override /model: ptg
- override /callbacks: default
- override /trainer: gpu
- override /paths: default
- override /logger: aim
# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters
exp_name: "debug"
tags: ["coffee", "ms_tcn", "debug"]
seed: 12345
trainer:
min_epochs: 50
max_epochs: 50
model:
compile: false
data:
batch_size: 1024
num_workers: 12
epoch_length: 20000
window_size: 75
paths:
data_dir: "/data/users/hannah.defazio/ptg_nas/data_copy/TCN_data/coffee_conf_10_hands_dist"
root_dir: "/data/PTG/cooking/training/activity_classifier/TCN_HPL"
logger:
aim:
experiment: ${exp_name}
capture_terminal_logs: true
task_name: ${exp_name}