nerfbaselines.training

class nerfbaselines.training.Trainer(*, train_dataset: ~nerfbaselines.types.Dataset, test_dataset: ~nerfbaselines.types.Dataset | None = None, method: ~nerfbaselines.types.Method, output: str = '.', save_iters: ~nerfbaselines.utils.Indices = 10000::10000, eval_few_iters: ~nerfbaselines.utils.Indices = ::2000, eval_all_iters: ~nerfbaselines.utils.Indices = -1, loggers: ~typing.FrozenSet[str] = frozenset({}), generate_output_artifact: bool | None = None, config_overrides: ~typing.Dict[str, ~typing.Any] | None = None)

Bases: object

eval_all()

eval_few()

get_logger() → Logger

property num_iterations

save()

train()

train_iteration()

nerfbaselines.training.eval_all(method: Method, logger: Logger | None, dataset: Dataset, *, output: str, step: int, evaluation_protocol: EvaluationProtocol, split: str, nb_info)

nerfbaselines.training.eval_few(method: Method, logger: Logger, dataset: Dataset, *, split: str, step, evaluation_protocol: EvaluationProtocol)