Commit d6d6e0b4 authored by Torge Berckmann's avatar Torge Berckmann
Browse files

Merge branch 'repr_compare' into local_master

parents 83f4478e a2d9baaa
......@@ -1087,6 +1087,7 @@ def dualinputs2ttransformer_base(args):
args.encoder_shared_layers = getattr(args, "encoder_shared_layers", 0)
args.decoder_layers = getattr(args, "decoder_layers", 6)
args.measure_corr = getattr(args, "measure_corr", False)
args.add_speech_eos = getattr(args, "add_speech_eos", False)
......
......@@ -372,7 +372,7 @@ class SpeechTextJointToTextTask(SpeechToTextTask):
return epoch_iter
def begin_valid_epoch(self, epoch, model):
if model.deoder.run_corr_arr:
if model.decoder.run_corr_arr:
corr_val = model.decoder.retrieve_corr()
msg = {
"epoch": epoch,
......@@ -382,7 +382,7 @@ class SpeechTextJointToTextTask(SpeechToTextTask):
model.decoder.reset_corr()
def post_validate(self, model, stats, agg):
if model.deoder.run_corr_arr:
if model.decoder.run_corr_arr:
corr_val = model.decoder.retrieve_corr()
msg = {
"correlation": corr_val
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment