Replies: 2 comments
-
Maybe you didn't setup device agnostic code for M1 chip GPU. |
Beta Was this translation helpful? Give feedback.
-
I have chosen to only use CPU and work in PyCharm. On a CPU only it does take a while to run.
|
Beta Was this translation helpful? Give feedback.
Uh oh!
There was an error while loading. Please reload this page.
-
Hello all,
I am now following the forth chapter and I'n getting very long train/test runtimes for model_0 (TinyVGG).
The example in the book ran in 4.935 seconds (for 5 epochs), yet on my computer each epoch takes 40-70 seconds!
I know there's a difference in the hardware (I'm running the code on my Mac M1 and not Colab+GPU) but up until now there wasn't such a difference in the runtimes.
Did anybody else encountered this issue?
[I pulled the notebook from this repo and ran it as-is, so this can't be a typo/coding error on my side]
P.S
I tried timing the test step in more detail, and it seems that most of the runtime is coming from the loop retrieving the values from the dataloader iterator. In addition, running the model on a single example image is very fast, so I really don't think this is a computational issue (at least not from the model itself).
Beta Was this translation helpful? Give feedback.
All reactions