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

OTapio March 25, 2020, 12:12pm 3

Thanks! All give that a shot.

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