From eec84855606dfad8aad8a927486b93900b018d47 Mon Sep 17 00:00:00 2001 From: Jirka Date: Fri, 5 Nov 2021 23:18:30 +0100 Subject: [PATCH] typos --- README.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/README.md b/README.md index 2d18bb8848..743a88c13b 100644 --- a/README.md +++ b/README.md @@ -209,7 +209,7 @@ model = ImageClassifier(backbone="resnet18", num_classes=2, optimizer="Adam", lr model = ImageClassifier(backbone="resnet18", num_classes=2, optimizer="Adam", lr_scheduler=functools.partial(CyclicLR, step_size_up=1500, mode='exp_range', gamma=0.5)) # - Tuple[string, dict]: (The dict takes in the scheduler kwargs) -model = ImageClassifier(backbone="resnet18", num_classes=2, optimizer="Adam", lr_scheduler=("StepLR", {"step_size": 10])) +model = ImageClassifier(backbone="resnet18", num_classes=2, optimizer="Adam", lr_scheduler=("StepLR", {"step_size": 10})) ``` You can also register you own custom scheduler recipes beforeahand and use them shown as above: @@ -238,10 +238,8 @@ The example also uses our [`merge_transforms`](https://lightning-flash.readthedo ```py import torch -from typing import Any import numpy as np import albumentations -from torchvision import transforms as T from flash.core.data.transforms import ApplyToKeys, merge_transforms from flash.image import ImageClassificationData from flash.image.classification.transforms import default_transforms, AlbumentationsAdapter