@TimedCall(seconds=200, daemon=False)
def test_training(self):
    torch.manual_seed(0)
    finish_state = run_training_test(self.data_dir, device=self.device)
    self.assertEqual(finish_state.iteration, 100)
    self.assertEqual(finish_state.epoch, 5)
@TimedCall(seconds=200, daemon=False)
def test_training(self):
    torch.manual_seed(0)
    finish_state = run_training_test(self.data_dir, device=self.device)
    self.assertEqual(finish_state.iteration, 100)
    self.assertEqual(finish_state.epoch, 5)