| | |
| | | assert check_argument_types() |
| | | if batch_size > 1: |
| | | raise NotImplementedError("batch decoding is not implemented") |
| | | if ngpu > 1: |
| | | raise NotImplementedError("only single GPU decoding is supported") |
| | | |
| | | |
| | | logging.basicConfig( |
| | | level=log_level, |
| | |
| | | device = "cuda" |
| | | else: |
| | | device = "cpu" |
| | | |
| | | batch_size = 1 |
| | | # 1. Set random-seed |
| | | set_all_random_seed(seed) |
| | | |
| | |
| | | **kwargs, |
| | | ): |
| | | assert check_argument_types() |
| | | if batch_size > 1: |
| | | raise NotImplementedError("batch decoding is not implemented") |
| | | if ngpu > 1: |
| | | raise NotImplementedError("only single GPU decoding is supported") |
| | | |
| | | |
| | | logging.basicConfig( |
| | | level=log_level, |
| | |
| | | device = "cuda" |
| | | else: |
| | | device = "cpu" |
| | | batch_size = 1 |
| | | |
| | | # 1. Set random-seed |
| | | set_all_random_seed(seed) |