modelzoo.common.pytorch.run_gpt_inference

GPT Inference script built using the cstorch API

Functions

get_cluster_config
get_parser
inference_input_dataloader
main

previous

modelzoo.common.pytorch.run_cstorch_flow.GradScalerParams

next

modelzoo.common.pytorch.run_gpt_inference.get_cluster_config