Skip to content

Commit 566e207

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b6a83da commit 566e207

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

2d_classification/monai_101.ipynb

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -264,10 +264,7 @@
264264
" network=model,\n",
265265
" inferer=SimpleInferer(),\n",
266266
" key_val_metric={\"val_acc\": ignite.metrics.Accuracy(from_engine([\"pred\", \"label\"]))},\n",
267-
" val_handlers=[\n",
268-
" StatsHandler(iteration_log=False),\n",
269-
" TensorBoardStatsHandler(iteration_log=False)\n",
270-
" ],\n",
267+
" val_handlers=[StatsHandler(iteration_log=False), TensorBoardStatsHandler(iteration_log=False)],\n",
271268
")\n",
272269
"trainer = SupervisedTrainer(\n",
273270
" device=torch.device(\"cuda:0\"),\n",
@@ -280,7 +277,7 @@
280277
" train_handlers=[\n",
281278
" ValidationHandler(validator=evaluator, epoch_level=True, interval=1),\n",
282279
" StatsHandler(),\n",
283-
" TensorBoardStatsHandler(tag_name=\"train_loss\", output_transform=from_engine([\"loss\"], first=True))\n",
280+
" TensorBoardStatsHandler(tag_name=\"train_loss\", output_transform=from_engine([\"loss\"], first=True)),\n",
284281
" ],\n",
285282
")"
286283
]

0 commit comments

Comments
 (0)