Network size should match qlty_window #33
Description
In train.py, the neural network is currently defined with the size of the tiled data. This should be updated such that it uses the size of the qlty patches instead while retaining the number of channels according to the input data.
As a test case, you can consider a data size of (217, 217) with qlty patches on (32, 32).