Replies: 1 comment
-
|
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
hello, mrdbourke
this is the code around at around 15:12:00
`from torch.utils.data import DataLoader
BATCH_SIZE = 32
train_dataloader = DataLoader(dataset=train_data, batch_size=BATCH_SIZE, shuffle=True)
test_dataloader = DataLoader(dataset=test_data, batch_size=BATCH_SIZE, shuffle=True)
train_dataloader, test_dataloader`
NOTICE: I have set the shuffle=True of test_dataloader which you set the value False
then I get the confmat_tensor matrix like this at around 19:24:00:
here is my doubts:
Epoch: 0
Train Loss: 0.5883, Train Acc:78.78%
Test Loss: 0.3993, Test Acc: 86.15%
Epoch: 1
Train Loss: 0.3657, Train Acc:86.88%
Test Loss: 0.3494, Test Acc: 87.49%
Epoch: 2
Train Loss: 0.3295, Train Acc:88.10%
Test Loss: 0.3224, Test Acc: 88.19%
Train time on None: 37.60443163400001 seconds
37.60443163400001
Beta Was this translation helpful? Give feedback.
All reactions