This commit is contained in:
Mayeul Aubin 2025-07-03 16:07:22 +02:00
parent d39cd7b1da
commit 75a3e5151e

View file

@ -208,7 +208,7 @@ class MomentaDataset(Dataset):
os.path.join(self.root_dir, self.INITIAL_CONDITIONS_DIR, f'ICs_{ID}_{var}.h5')
for var in self.initial_conditions_variables
]
target_path = os.path.join(self.root_dir, self.TARGET_DIR, f'{self.target_variable}_{ID}_nforce{t}.h5')
target_path = os.path.join(self.root_dir, self.TARGET_DIR, f'{self.target_variable}_{ID}_{t}.h5')
style_path = os.path.join(self.root_dir, self.STYLE_DIR, f'{self.style_files}_{ID}_nforce{t}.txt')
# Read 3D chunks