From 79af8d9db0cce42109287393be1ac0229b693a24 Mon Sep 17 00:00:00 2001 From: lmeribal Date: Wed, 11 Sep 2024 20:44:54 +0300 Subject: [PATCH] in progress --- turbo_alignment/pipelines/preprocessing/multimodal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/turbo_alignment/pipelines/preprocessing/multimodal.py b/turbo_alignment/pipelines/preprocessing/multimodal.py index 81b7156..d9ea1fa 100644 --- a/turbo_alignment/pipelines/preprocessing/multimodal.py +++ b/turbo_alignment/pipelines/preprocessing/multimodal.py @@ -101,7 +101,7 @@ def _async_process_files(self, reader, encoder, files_paths, experiment_settings logger.info(f'📖 Encoding batch {i} / {len(batches)}') batch_output = self._process_function(reader, encoder, batch, experiment_settings, i) torch.save(batch_output, experiment_settings.output_file_path / ( - 'process_' + str(self.accelerator.process_index) + 'batch_' + str(i) + '.' + experiment_settings.modality.value + 'process_' + str(self.accelerator.process_index) + '_batch_' + str(i) + '.' + experiment_settings.modality.value + '.' + experiment_settings.encoder_settings.modality_encoder_type + '.pt'