GPU vs CPU performance #1195
sohanasarah
started this conversation in
General
Replies: 1 comment
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I am training neural-prophet with a large dataset. I am trying to use accelerator: gpu feature but I noticed that training performance is better on cpu than gpu. I tried to increase batch size and decrease epoch still I get the same result. Am I missing something? Below is the parameter setup that I used:
params = dict(
weekly_seasonality = False,
daily_seasonality = False,
seasonality_mode = 'multiplicative',
epochs=500,
learning_rate=0.05,
accelerator = "gpu"
)
Beta Was this translation helpful? Give feedback.
All reactions