generated from ashleve/lightning-hydra-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrn_val_tst.yaml
35 lines (32 loc) · 962 Bytes
/
trn_val_tst.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
# @package _global_
# specify here defaults
defaults:
- _self_
- experiment: trn_val_tst # Global parameters of experiment
- datamodule: tabular # Information about dataset
- trainer: gpu # Run configuration for DNN models
- callbacks: default # Callbacks for DNN models
- logger: none # Loggers for DNN models
- hydra: default.yaml # Output paths for logs
# Models options:
# Linear:
# elastic_net
# GBDT:
# xgboost
# lightgbm
# catboost
# DNN:
# widedeep_tab_mlp
# widedeep_tab_net
# widedeep_ft_transformer
# widedeep_saint
# pytorch_tabular_autoint
# pytorch_tabular_tabnet
# pytorch_tabular_node
# nam
# danet
- model: widedeep_ft_transformer
- hparams_search: widedeep_ft_transformer # Model-specific hyperparameters
# enable color logging
- override hydra/hydra_logging: colorlog
- override hydra/job_logging: colorlog