modelzoo.common.pytorch.run_cstorch_flow.log_input_summary

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

Log the input tensors to tensorboard

previous

modelzoo.common.pytorch.run_cstorch_flow.get_model_checkpoint

next

modelzoo.common.pytorch.run_cstorch_flow.optimizer_step_with_summaries