gin/train/train_newt.gin

get_model.model = @NeuralWaveshaping

include 'gin/models/newt.gin'

URMPDataModule.batch_size = 8

NeuralWaveshaping.learning_rate = 0.001
NeuralWaveshaping.lr_decay = 0.9
NeuralWaveshaping.lr_decay_interval = 10000

trainer_kwargs.max_steps = 120000
trainer_kwargs.gradient_clip_val = 2.0
trainer_kwargs.accelerator = 'dp'