neuraltrain.utils.BaseExperiment¶

class neuraltrain.utils.BaseExperiment(*, infra: TaskInfra = TaskInfra(folder=None, cluster=None, logs='{folder}/logs/{user}/%j', job_name=None, timeout_min=None, nodes=1, tasks_per_node=1, cpus_per_task=None, gpus_per_node=None, mem_gb=None, max_pickle_size_gb=None, slurm_constraint=None, slurm_partition=None, slurm_account=None, slurm_qos=None, slurm_use_srun=False, slurm_additional_parameters=None, conda_env=None, workdir=None, permissions=511, version='0', mode='cached', keep_in_ram=False))[source][source]¶: Base experiment class which require an infra and a ‘run’ method.