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
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
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