class

GS::GaussianSplatting::Trainer

Inherits Reference < Object

Main trainer class

Constructors

load_checkpoint(path : String) : Trainer

Load checkpoint

Source
new(gaussians : Gaussian3D, cameras : CameraSet, config : TrainerConfig = TrainerConfig.new)
Source

Instance methods

cameras
Source
cameras=(cameras : CameraSet)
Source
config
Source
config=(config : TrainerConfig)
Source
gaussians
Source
gaussians=(gaussians : Gaussian3D)
Source
save_checkpoint(path : String) : Nil

Save checkpoint

Source
train

Run full training

Source
train_step

Single training step

Source