From 8ee1e53fee09bf7ab75c9d167319cc6ae77af8c8 Mon Sep 17 00:00:00 2001 From: jayLEE0301 Date: Fri, 24 May 2024 14:29:00 -0400 Subject: [PATCH] remove redundant comment, change scheduler name --- lerobot/common/policies/vqbet/modeling_vqbet.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/lerobot/common/policies/vqbet/modeling_vqbet.py b/lerobot/common/policies/vqbet/modeling_vqbet.py index 1de6d79c..437d04e6 100644 --- a/lerobot/common/policies/vqbet/modeling_vqbet.py +++ b/lerobot/common/policies/vqbet/modeling_vqbet.py @@ -490,10 +490,9 @@ class VQBeTScheduler: def __init__(self, optimizer, cfg): from diffusers.optimization import get_scheduler self.discretize_step = cfg.training.discretize_step - # self.offline_steps = cfg.training.offline_steps self.optimizing_step = 0 - self.lr_scheduler1 = get_scheduler( + self.lr_scheduler = get_scheduler( cfg.training.lr_scheduler, optimizer=optimizer.encoder_optimizer, num_warmup_steps=cfg.training.lr_warmup_steps, @@ -504,9 +503,7 @@ class VQBeTScheduler: def step(self): self.optimizing_step +=1 if self.optimizing_step >= self.discretize_step: - self.lr_scheduler1.step() - # self.lr_scheduler2.step() - # self.lr_scheduler3.step() + self.lr_scheduler.step() class VQBeTRgbEncoder(nn.Module): """Encoder an RGB image into a 1D feature vector.