Offline Datasets res net optionality

2020-03-12 18:32:23 +01:00
parent 2f99341cc3
commit bb47e07566
11 changed files with 638 additions and 140 deletions
--- a/lib/preprocessing/generator.py
+++ b/lib/preprocessing/generator.py
@ -20,6 +20,8 @@ class Generator:

        self.data_root = Path(data_root)

+
+
    def generate_n_trajectories_m_alternatives(self, n, m, datafile_name, processes=0, **kwargs):
        datafile_name = datafile_name if datafile_name.endswith('.pik') else f'{str(datafile_name)}.pik'
        kwargs.update(n=m)