cerebras.modelzoo.common.run_cstorch_flow.log_input_summary

cerebras.modelzoo.common.run_cstorch_flow.log_input_summary(batch)

Log the input tensors to tensorboard

previous

cerebras.modelzoo.common.run_cstorch_flow.get_model_checkpoint

next

cerebras.modelzoo.common.run_cstorch_flow.optimizer_step_with_summaries