Olamide Isreal
a3ffec6a07
Some checks failed
Code Quality Main / code-quality (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.10) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.8) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.9) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.10) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.8) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.9) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.10) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.8) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.9) (push) Has been cancelled
Tests / code-coverage (push) Has been cancelled
36 lines
760 B
YAML
36 lines
760 B
YAML
# @package _global_
|
|
|
|
# to execute this experiment run:
|
|
# python train.py experiment=flowdock_fm
|
|
|
|
defaults:
|
|
- override /data: combined
|
|
- override /model: flowdock_fm
|
|
- override /callbacks: default
|
|
- override /trainer: default
|
|
|
|
# all parameters below will be merged with parameters from default configurations set above
|
|
# this allows you to overwrite only specified parameters
|
|
|
|
tags: ["flowdock_fm", "combined_dataset"]
|
|
|
|
seed: 496
|
|
|
|
trainer:
|
|
max_epochs: 300
|
|
check_val_every_n_epoch: 5 # NOTE: we increase this since validation steps involve full model sampling and evaluation
|
|
reload_dataloaders_every_n_epochs: 1
|
|
|
|
model:
|
|
optimizer:
|
|
lr: 2e-4
|
|
compile: false
|
|
|
|
data:
|
|
batch_size: 16
|
|
|
|
logger:
|
|
wandb:
|
|
tags: ${tags}
|
|
group: "FlowDock-FM"
|