added in_memory loading to reduce disk reads and increase speed

Add code so loss function uses torch.Size([x,1]) instead of torch.Size([x])

add testing notebook for convolutional neural network

add FNN with MSE loss

add missing changes from original repository

cleanup to make ready as submodule