def test_epoch_continuous_data(self):
mono = model.Monokuma(len(self.dat_cont.i2w),
self.dim,
self.num_layers,
self.dropout,
is_conditional=False,
bidirectional_encoder=False,
use_bridge=False,
use_attention=False)
junko = control.Junko(mono, self.lr, self.bptt, self.interval)
val_loss, _, epoch_time = junko.epoch_continuous_data(self.dat_cont, 0)
def test_epoch_translation_data_simple(self):
mono = model.Monokuma(len(self.dat_tran.i2w),
self.dim,
self.num_layers,
self.dropout,
is_conditional=True,
bidirectional_encoder=False,
use_bridge=False,
use_attention=False)
junko = control.Junko(mono, self.lr, self.bptt, self.interval)
val_loss, _, epoch_time = junko.epoch_translation_data(self.dat_tran,
0,
shuffle=False)