Skip to content

Commit

Permalink
Revert deprecated changes
Browse files Browse the repository at this point in the history
Signed-off-by: Chaurasiya, Payal <[email protected]>
  • Loading branch information
payalcha committed Dec 11, 2024
1 parent 51d8744 commit d1517c7
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -135,8 +135,10 @@
"source": [
"from torch.utils.tensorboard import SummaryWriter\n",
"\n",
"writer = SummaryWriter('./logs/cnn_mnist', flush_secs=5)\n",
"\n",
"\n",
"def write_metric(node_name, task_name, metric_name, metric, round_number):\n",
" writer = SummaryWriter('./logs/cnn_mnist', flush_secs=5)\n",
" writer.add_scalar(\"{}/{}/{}\".format(node_name, task_name, metric_name),\n",
" metric, round_number)"
]
Expand Down

0 comments on commit d1517c7

Please sign in to comment.