dacapo.experiments.run

Module Contents

Classes

Run

class dacapo.experiments.run.Run(run_config)
name: str
train_until: int
validation_interval: int
task: dacapo.experiments.tasks.task.Task
architecture: dacapo.experiments.architectures.Architecture
trainer: dacapo.experiments.trainers.Trainer
datasplit: dacapo.experiments.datasplits.DataSplit
model: dacapo.experiments.model.Model
optimizer: torch.optim.Optimizer
training_stats: dacapo.experiments.training_stats.TrainingStats
validation_scores: dacapo.experiments.validation_scores.ValidationScores
static get_validation_scores(run_config) dacapo.experiments.validation_scores.ValidationScores

Static method to avoid having to initialize model, optimizer, trainer, etc.

move_optimizer(device: torch.device, empty_cuda_cache: bool = False) None