learn-deeply t1_itikofd wrote on October 23, 2022 at 10:17 PM

Reply to comment by BITE_AU_CHOCOLAT in [D] Building the Future of TensorFlow by eparlan

PyTorch doesn't inherently use more or less memory than TensorFlow, there's a bug in your code. If it's easier to switch frameworks than debug, more power to you.

BITE_AU_CHOCOLAT t1_itiofjz wrote on October 23, 2022 at 10:43 PM

Well, I haven't "switched" since I've been using Tensorflow since the start of the project. I was just curious to see if Pytorch could allow me to squeeze more juice and after spending a weekend trying to learn ~~assembly~~ Pytorch syntax it turns out that yes, but actually no. So yeah I'm perfectly content with using model.fit and calling it a day for the time being.

Oh and I also forgot: Pytorch won't train with a distributed strategy in a Jupyter environment. KEK.