Skip to content

Commit 1d4b0ce

Browse files
authored
NuClick needed to adapt the latest changes to AsDiscreted transform input arguments and other minor path fixes in SSL (#924)
Signed-off-by: Vishwesh Nath <vnath@nvidia.com> Signed-off-by: Vishwesh Nath <vnath@nvidia.com> Co-authored-by: Vishwesh Nath <vnath@nvidia.com>
1 parent 0ab98bc commit 1d4b0ce

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

nuclick/nuclick_training_notebook.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2278,7 +2278,7 @@
22782278
"\n",
22792279
"train_post_transforms = Compose([\n",
22802280
" Activationsd(keys=\"pred\", sigmoid=True),\n",
2281-
" AsDiscreted(keys=\"pred\", threshold_values=True, logit_thresh=0.5),\n",
2281+
" AsDiscreted(keys=\"pred\", threshold=0.5),\n",
22822282
"]\n",
22832283
")\n",
22842284
"\n",
@@ -2507,9 +2507,9 @@
25072507
],
25082508
"metadata": {
25092509
"kernelspec": {
2510-
"display_name": "nuclick_tutorial",
2510+
"display_name": "py38_monai_v1_tutorial_test",
25112511
"language": "python",
2512-
"name": "nuclick_tutorial"
2512+
"name": "py38_monai_v1_tutorial_test"
25132513
},
25142514
"language_info": {
25152515
"codemirror_mode": {

self_supervised_pretraining/ssl_script_train.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727
def main():
2828

2929
#TODO Defining file paths & output directory path
30-
json_Path = os.path.normpath('/scratch/data_2021/tcia_covid19/dataset_split_debug.json')
31-
data_Root = os.path.normpath('/scratch/data_2021/tcia_covid19')
32-
logdir_path = os.path.normpath('/home/vishwesh/monai_tutorial_testing/issue_467')
30+
json_Path = os.path.normpath('/to/be/defined/by/user/')
31+
data_Root = os.path.normpath('/to/be/defined/by/user/')
32+
logdir_path = os.path.normpath('/to/be/defined/by/user/')
3333

3434
if os.path.exists(logdir_path)==False:
3535
os.mkdir(logdir_path)

0 commit comments

Comments
 (0)