From ba5c0587f5aa6a20ff31c89ce50e2889835bac12 Mon Sep 17 00:00:00 2001 From: Siam Arefin Date: Sat, 14 Dec 2024 22:39:09 +0600 Subject: [PATCH 1/5] Bug fix --- Seg_UKAN/requirements.txt | 21 +++++++++------------ Seg_UKAN/train.py | 2 +- 2 files changed, 10 insertions(+), 13 deletions(-) diff --git a/Seg_UKAN/requirements.txt b/Seg_UKAN/requirements.txt index 3c14a7c..ca31ca9 100644 --- a/Seg_UKAN/requirements.txt +++ b/Seg_UKAN/requirements.txt @@ -1,20 +1,17 @@ -addict==2.4.0 +addict dataclasses pandas pyyaml albumentations tqdm tensorboardX -# mmcv-full==1.2.7 numpy opencv-python -perceptual==0.1 -pillow==8.4.0 -scikit-image==0.17.2 -scipy==1.5.4 -tifffile==2020.9.3 -timm==0.3.2 -typing-extensions==4.0.0 -yapf==0.31.0 - -pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116 \ No newline at end of file +perceptual +pillow +scikit-image +scipy +tifffile +timm +typing-extensions +yapf \ No newline at end of file diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py index fd5ee00..0eb98dc 100644 --- a/Seg_UKAN/train.py +++ b/Seg_UKAN/train.py @@ -39,7 +39,7 @@ from pdb import set_trace as st -ARCH_NAMES = archs.__all__ +# ARCH_NAMES = archs.__all__ LOSS_NAMES = losses.__all__ LOSS_NAMES.append('BCEWithLogitsLoss') From 8d51d79db184e4616760ed56d0ae074a2e107011 Mon Sep 17 00:00:00 2001 From: Siam arefin Date: Sat, 14 Dec 2024 23:22:59 +0600 Subject: [PATCH 2/5] Update train.py --- Seg_UKAN/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py index 0eb98dc..846fd0d 100644 --- a/Seg_UKAN/train.py +++ b/Seg_UKAN/train.py @@ -55,7 +55,7 @@ def parse_args(): parser.add_argument('--name', default=None, help='model name: (default: arch+timestamp)') - parser.add_argument('--epochs', default=400, type=int, metavar='N', + parser.add_argument('--epochs', default=5, type=int, metavar='N', help='number of total epochs to run') parser.add_argument('-b', '--batch_size', default=8, type=int, metavar='N', help='mini-batch size (default: 16)') From 32a3b913f8c92c01bdca82cd04924f564d3011ee Mon Sep 17 00:00:00 2001 From: Siam arefin Date: Sat, 14 Dec 2024 23:32:58 +0600 Subject: [PATCH 3/5] Update train.py --- Seg_UKAN/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py index 846fd0d..0eb98dc 100644 --- a/Seg_UKAN/train.py +++ b/Seg_UKAN/train.py @@ -55,7 +55,7 @@ def parse_args(): parser.add_argument('--name', default=None, help='model name: (default: arch+timestamp)') - parser.add_argument('--epochs', default=5, type=int, metavar='N', + parser.add_argument('--epochs', default=400, type=int, metavar='N', help='number of total epochs to run') parser.add_argument('-b', '--batch_size', default=8, type=int, metavar='N', help='mini-batch size (default: 16)') From 035abe519264d150a74c106ccbd321d7df63a368 Mon Sep 17 00:00:00 2001 From: Siam arefin Date: Sun, 15 Dec 2024 00:33:57 +0600 Subject: [PATCH 4/5] Update train.py --- Seg_UKAN/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py index 0eb98dc..c33a419 100644 --- a/Seg_UKAN/train.py +++ b/Seg_UKAN/train.py @@ -55,7 +55,7 @@ def parse_args(): parser.add_argument('--name', default=None, help='model name: (default: arch+timestamp)') - parser.add_argument('--epochs', default=400, type=int, metavar='N', + parser.add_argument('--epochs', default=2, type=int, metavar='N', help='number of total epochs to run') parser.add_argument('-b', '--batch_size', default=8, type=int, metavar='N', help='mini-batch size (default: 16)') From 7caaf545ffa1ffba2942cafd0c6b1f873b444c05 Mon Sep 17 00:00:00 2001 From: Siam arefin Date: Sun, 15 Dec 2024 00:46:17 +0600 Subject: [PATCH 5/5] Update train.py --- Seg_UKAN/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py index c33a419..0eb98dc 100644 --- a/Seg_UKAN/train.py +++ b/Seg_UKAN/train.py @@ -55,7 +55,7 @@ def parse_args(): parser.add_argument('--name', default=None, help='model name: (default: arch+timestamp)') - parser.add_argument('--epochs', default=2, type=int, metavar='N', + parser.add_argument('--epochs', default=400, type=int, metavar='N', help='number of total epochs to run') parser.add_argument('-b', '--batch_size', default=8, type=int, metavar='N', help='mini-batch size (default: 16)')