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()
|