| | |
|---|
| | | output_dir: Optional[str] = None, |
|---|
| | | batch_size: int = 1, |
|---|
| | | dtype: str = "float32", |
|---|
| | | ngpu: int = 0, |
|---|
| | | ngpu: int = 1, |
|---|
| | | num_workers: int = 0, |
|---|
| | | log_level: Union[int, str] = "INFO", |
|---|
| | | key_file: Optional[str] = None, |
|---|
| | |
|---|
| | | output_writer = open("{}/result.txt".format(output_path), "w") |
|---|
| | | result_list = [] |
|---|
| | | for keys, batch in loader: |
|---|
| | | logger.info("keys: {}".format(keys)) |
|---|
| | | logger.info("batch: {}".format(batch)) |
|---|
| | | assert isinstance(batch, dict), type(batch) |
|---|
| | | assert all(isinstance(s, str) for s in keys), keys |
|---|
| | | _bs = len(next(iter(batch.values()))) |
|---|
