()
| 276 | f'Train/lamb_coeffs', lamb_coeffs, global_step) |
| 277 | |
| 278 | def get_arguments(): |
| 279 | parser = get_argument_parser() |
| 280 | # Include DeepSpeed configuration arguments |
| 281 | parser = deepspeed.add_config_arguments(parser) |
| 282 | |
| 283 | args = parser.parse_args() |
| 284 | |
| 285 | # no cuda mode is not supported |
| 286 | args.no_cuda = False |
| 287 | |
| 288 | return args |
| 289 | |
| 290 | def construct_arguments(): |
| 291 | args = get_arguments() |
no test coverage detected