I have been trying to create ResNet-18 on Julia. I have managed to create a functional one but it is slow and uses lot of GPU memory. This is puzzling since similar architecture on Tensorflow runs 5-10 times faster and uses much less memory. Also, there is a implementation of VGG-16, which I found t…

A implementation of ResNet-18 uses lot of GPU memory

CarloLucibello May 9, 2020, 5:50am 11

@OTapio could you open an issue on Flux?

Topic		Replies	Views
Memory challenges for Flux on Resnet Machine Learning gpu	8	1390	September 7, 2022
Flux runs out of memory Machine Learning memory-allocation , flux	25	4373	June 1, 2023
MNIST GPU CuArrays error GPU	23	3085	January 22, 2019
Memory usage increasing with each epoch Machine Learning cuda , flux	18	761	April 14, 2025
Flux Transformer Out of Memory Machine Learning	25	1524	March 13, 2023