Skip to content

Commit 18daff1

Browse files
Update losses doc (#6691)
Fixes #6682. ### Description Add DeepSupervisionLoss in the doc ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u--net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: HussainAlasmawi <[email protected]>
1 parent dcdae0d commit 18daff1

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

docs/source/losses.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,3 +112,8 @@ Loss Wrappers
112112
~~~~~~~~~~~~
113113
.. autoclass:: MaskedLoss
114114
:members:
115+
116+
`DeepSupervisionLoss`
117+
~~~~~~~~~~~~~~~~~~~~~
118+
.. autoclass:: DeepSupervisionLoss
119+
:members:

tests/test_vitautoenc.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
from monai.networks import eval_mode
1919
from monai.networks.nets.vitautoenc import ViTAutoEnc
20-
from tests.utils import skip_if_windows
20+
from tests.utils import skip_if_quick, skip_if_windows
2121

2222
TEST_CASE_Vitautoenc = []
2323
for in_channels in [1, 4]:
@@ -64,7 +64,8 @@
6464
)
6565

6666

67-
class TestPatchEmbeddingBlock(unittest.TestCase):
67+
@skip_if_quick
68+
class TestVitAutoenc(unittest.TestCase):
6869
def setUp(self):
6970
self.threads = torch.get_num_threads()
7071
torch.set_num_threads(4)

0 commit comments

Comments
 (0)