a_d:

Also here you are concatenating an empty tensor to your model output, I do not know what exactly is the overall model and objective, but this statement seems to have no effect.


the dataset is very large so i split it into batches of 4 items, then i concatenate the result of each batch with the other results. In this way i can use all the images in the dataset in each epoch.