FlenneSoyeux

Flux : slower and slower computations of a neural network - a VRAM problem ?

3

This small code: ``` input = rand(Float32, 6, 6, 2, 1) |> gpu output = nn(input) |> cpu ``` is slower and slower to compute on my computer. Namely, the...