gher-uliege / DINCAE

DINCAE (Data-Interpolating Convolutional Auto-Encoder) is a neural network to reconstruct missing data in satellite observations.
GNU General Public License v3.0
47 stars 19 forks source link

Ensure consistency of 'train' argument between data generator functions #19

Open HeQinWill opened 6 months ago

HeQinWill commented 6 months ago

Fixed a bug where the 'train' argument was not being consistently passed from the 'data_generator' function to the 'data_generator_list' function. This inconsistency was causing the 'train' value to unexpectedly switch to 'True' during data generation for the testing part. The 'train' argument is now directly passed from 'data_generator' to 'data_generator_list', ensuring its value remains consistent throughout the data generation process.