| 413 | |
| 414 | |
| 415 | class ArgHelper(object): |
| 416 | def __init__(self, model_type=None, data_type=None, decoder_ths_path=None, decoding_ths_path=None): |
| 417 | self.model_type = model_type |
| 418 | self.data_type = data_type |
| 419 | self.decoder_ths_path = decoder_ths_path |
| 420 | self.decoding_ths_path = decoding_ths_path |
| 421 | |
| 422 | |
| 423 | class TorchDecoding(nn.Module): |