Skip to content

Commit 30f3589

Browse files
authored
Update README.md
1 parent 2a9aa6d commit 30f3589

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,16 +41,16 @@ metric = paddle.metric.Accuracy(topk=(1, 5))
4141
model.prepare(optimizer=opt, loss=loss, metrics=metric)
4242

4343
# Set the data preprocess
44-
transforms = T.Compose([
44+
train_transforms = T.Compose([
4545
T.Resize(256, interpolation='bicubic'),
4646
T.RandomCrop(224),
4747
T.ToTensor(),
4848
T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
4949
])
5050

5151
# Load the Cifar100 dataset
52-
train_dataset = Cifar100(mode='train', transform=transforms, backend='pil')
53-
val_dataset = Cifar100(mode='test', transform=transforms, backend='pil')
52+
train_dataset = Cifar100(mode='train', transform=train_transforms, backend='pil')
53+
val_dataset = Cifar100(mode='test', transform=val_transforms, backend='pil')
5454

5555
# Finetune the model
5656
model.fit(

0 commit comments

Comments
 (0)