Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 9, 2024
1 parent 7b03936 commit 7f7db3e
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions modules/torch_compile.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -376,24 +376,24 @@
"outputs": [],
"source": [
"trainer = SupervisedTrainer(\n",
" device=device,\n",
" max_epochs=epoch_num,\n",
" train_data_loader=data_loader,\n",
" network=model,\n",
" # optimizer=optimizer,\n",
" # loss_function=loss_function,\n",
" # inferer=SimpleInferer(),\n",
" # postprocessing=post_transform,\n",
" # amp=args.amp,\n",
" # key_train_metric={\n",
" # \"train_dice\": MeanDice(\n",
" # include_background=False,\n",
" # output_transform=from_engine([\"pred\", \"label\"]),\n",
" # )\n",
" # },\n",
" compile=True,\n",
" # you can also add `compile_kwargs` dict of the args for `torch.compile()` API\n",
" compile_kwargs={}\n",
" device=device,\n",
" max_epochs=epoch_num,\n",
" train_data_loader=data_loader,\n",
" network=model,\n",
" # optimizer=optimizer,\n",
" # loss_function=loss_function,\n",
" # inferer=SimpleInferer(),\n",
" # postprocessing=post_transform,\n",
" # amp=args.amp,\n",
" # key_train_metric={\n",
" # \"train_dice\": MeanDice(\n",
" # include_background=False,\n",
" # output_transform=from_engine([\"pred\", \"label\"]),\n",
" # )\n",
" # },\n",
" compile=True,\n",
" # you can also add `compile_kwargs` dict of the args for `torch.compile()` API\n",
" compile_kwargs={},\n",
")"
]
},
Expand Down

0 comments on commit 7f7db3e

Please sign in to comment.