cerebras.modelzoo.common.run_cstorch_flow.run_cstorch_eval#

cerebras.modelzoo.common.run_cstorch_flow.run_cstorch_eval(params, params_obj, model_fn, input_fn, cluster_config, artifact_dir)[source]#

Runs the evaluation workflow built using the cstorch API.

Parameters

  • params – the params dictionary extracted from the params.yaml used

  • model_fn – A callable that takes in the params dictionary and returns a torch.nn.Module

  • input_data_fn – A callable that takes in the param dictionary and returns a torch.utils.data.DataLoader

previous

cerebras.modelzoo.common.run_cstorch_flow.optimizer_step_with_summaries

next

cerebras.modelzoo.common.run_cstorch_flow.run_cstorch_flow