Dev gzf train (#1524)
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* train (#1521) (#1522)
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
* trainer
| | |
| | | hlens: torch.Tensor, |
| | | ys_in_pad: torch.Tensor, |
| | | ys_in_lens: torch.Tensor, |
| | | return_hidden: bool = False, |
| | | return_both: bool= False, |
| | | chunk_mask: torch.Tensor = None, |
| | | return_hidden: bool = False, |
| | | return_both: bool = False, |
| | | ) -> Tuple[torch.Tensor, torch.Tensor]: |
| | | """Forward decoder. |
| | | |
| | |
| | | import os |
| | | import io |
| | | import shutil |
| | | import logging |
| | | from collections import OrderedDict |
| | | import numpy as np |
| | | from omegaconf import DictConfig, OmegaConf |