BaseTrainer
-
class
packnet_sfm.trainers.base_trainer.BaseTrainer(min_epochs=0, max_epochs=50, checkpoint=None, **kwargs)[source] Bases:
object-
check_and_save(module, output)[source]
-
property
is_rank_0
-
property
proc_rank
-
test_progress_bar(dataloader, config, n=0, ncols=120)[source]
-
train_progress_bar(dataloader, config, ncols=120)[source]
-
val_progress_bar(dataloader, config, n=0, ncols=120)[source]
-
property
world_size
-
-
packnet_sfm.trainers.base_trainer.sample_to_cuda(data, dtype=None)[source]