Skip to content

Commit 9a2fc55

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 24ddbd0 commit 9a2fc55

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

2d_classification/MIL/mil_classify_monai.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def __init__(self, num_classes, mil_mode='att', pretrained=True, backbone=None,
135135
n_trans=4
136136
trans_dropout=0.0
137137
self.attention = None
138-
138+
139139

140140
if backbone is None:
141141
#use the ResNet50 backbone
@@ -489,8 +489,8 @@ def __call__(self, data):
489489
class LabelEncodeIntegerGraded(transforms.Transform):
490490
"""
491491
Convert to integer label to encoded array representation of length num_classes,
492-
with 1 filled in up to label index, and 0 otherwise. For example for num_classes=5,
493-
embedding of 2 -> (1,1,0,0,0)
492+
with 1 filled in up to label index, and 0 otherwise. For example for num_classes=5,
493+
embedding of 2 -> (1,1,0,0,0)
494494
495495
Args:
496496
num_classes: the number of classes to convert to encoded format.
@@ -941,4 +941,4 @@ def main_worker(gpu, args):
941941
print('ALL DONE')
942942

943943
if __name__ == '__main__':
944-
main()
944+
main()

0 commit comments

Comments
 (0)