(output_path, args)
| 28 | continue |
| 29 | |
| 30 | def json_dumps_arguments(output_path, args): |
| 31 | with open(output_path, "w") as f: |
| 32 | params = vars(args) |
| 33 | if "device" in params: |
| 34 | params["device"] = str(params["device"]) |
| 35 | f.write(json.dumps(params, indent=4)) |
| 36 | |
| 37 | |
| 38 | def split_and_padding_neighbor(batch_tensor, batch_len): |