search

phlippe / uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
https://uvadlc-notebooks.readthedocs.io/en/latest/
MIT License
2.31k stars 541 forks source link

JAX Tutorial 5 DenseNet Memory Access Violation Error #98

Closed foainla closed 1 year ago

foainla commented 1 year ago

Tutorial: 5

When I try to run the DenseNet model with bn_size : int = 4, growth_rate : int = 32, I get "Process finished with exit code -1073741819 (0xC0000005)" I get the same error if I try to num_layers : tuple = (6, 12, 48, 32), bn_size : int = 2, growth_rate : int = 16

Runtime environment

Additional context My overall goal is to recreate DenseNet201 with num_layers : tuple = (6, 12, 48, 32), bn_size : int = 4, growth_rate : int = 32

phlippe commented 1 year ago

I am not sure if this is due to the provided code or a potential issue in the installation. Can you check that JAX can access your GPU as normal and has enough GPU memory to run a small model?

phlippe commented 1 year ago

Closing due to inactivity. If the issue remains, feel free to reopen!