Solved by albanD in post #2 
              
              
                Hi, 
I don’t think there is any reason for the autograd to fail here. 
You might want to make sure that you preprocess your data properly though to avoid any large value as it could hinder training. 
For the architecture, it is very task-dependent I’m afraid. But for vision related tasks a pre-train…