|
@@ -126,7 +126,7 @@ class GaussianModel:
|
|
|
self.xyz_scheduler_args = get_expon_lr_func(lr_init=training_args.position_lr_init*self.spatial_lr_scale,
|
|
self.xyz_scheduler_args = get_expon_lr_func(lr_init=training_args.position_lr_init*self.spatial_lr_scale,
|
|
|
lr_final=training_args.position_lr_final*self.spatial_lr_scale,
|
|
lr_final=training_args.position_lr_final*self.spatial_lr_scale,
|
|
|
lr_delay_mult=training_args.position_lr_delay_mult,
|
|
lr_delay_mult=training_args.position_lr_delay_mult,
|
|
|
- max_steps=training_args.posititon_lr_max_steps)
|
|
|
|
|
|
|
+ max_steps=training_args.position_lr_max_steps)
|
|
|
|
|
|
|
|
def update_learning_rate(self, iteration):
|
|
def update_learning_rate(self, iteration):
|
|
|
''' Learning rate scheduling per step '''
|
|
''' Learning rate scheduling per step '''
|