Merge pull request #926 from wannature/singa_v3
update comments for class Compose
diff --git a/examples/demos/Classification/BloodMnist/transforms.py b/examples/demos/Classification/BloodMnist/transforms.py
index 59dc099..54d9d12 100644
--- a/examples/demos/Classification/BloodMnist/transforms.py
+++ b/examples/demos/Classification/BloodMnist/transforms.py
@@ -26,33 +26,6 @@
class Compose:
- """Composes several transforms together. This transform does not support torchscript.
- Please, see the note below.
-
- Args:
- transforms (list of ``Transform`` objects): list of transforms to compose.
-
- Example:
- >>> transforms.Compose([
- >>> transforms.CenterCrop(10),
- >>> transforms.PILToTensor(),
- >>> transforms.ConvertImageDtype(torch.float),
- >>> ])
-
- .. note::
- In order to script the transformations, please use ``torch.nn.Sequential`` as below.
-
- >>> transforms = torch.nn.Sequential(
- >>> transforms.CenterCrop(10),
- >>> transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
- >>> )
- >>> scripted_transforms = torch.jit.script(transforms)
-
- Make sure to use only scriptable transformations, i.e. that work with ``torch.Tensor``, does not require
- `lambda` functions or ``PIL.Image``.
-
- """
-
def __init__(self, transforms):
self.transforms = transforms