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	1370	September 7, 2022
Flux + GPU memory problems Machine Learning flux	2	816	April 26, 2022
Flux memory usage high in SRCNN Machine Learning	3	192	June 5, 2024
Flux runs out of memory Machine Learning memory-allocation , flux	25	4295	June 1, 2023
Memory usage increasing with each epoch Machine Learning cuda , flux	18	698	April 14, 2025