CNN works with other implementation

[AntoAndGar]

Just to notice, I have readapted your code here: https://github.com/AntoAndGar/Intrinsic-Dimension and have found that CNN models work but you have a different implementation from the main paper so the results are different.