6 lines
191 B
Python
6 lines
191 B
Python
from omegaconf import OmegaConf
|
|
from rnn_trainer import BrainToTextDecoder_Trainer
|
|
|
|
args = OmegaConf.load('rnn_args.yaml')
|
|
trainer = BrainToTextDecoder_Trainer(args)
|
|
metrics = trainer.train() |