About train_util.py

[Devil-Ideal]

During training, the while loop in the 'run_loop' function does not seem to terminate because self. lr_ annel_ Steps=0, not self. lr_ annel_ Steps are always true.

the code is below:
segmentation_train.py
def create_argparser():
defaults = dict(
data_dir="./data/training",
schedule_sampler="uniform",
lr=1e-4,
weight_decay=0.0,
lr_anneal_steps=0,
batch_size=1,
microbatch=-1, # -1 disables microbatches
ema_rate="0.9999", # comma-separated list of EMA values
log_interval=100,
save_interval=5000,
resume_checkpoint='',#'"./results/pretrainedmodel.pt",
use_fp16=False,
fp16_scale_growth=1e-3,
)
train_util.py
class TrainLoop:
def init(
self,
*,
model,
classifier,
diffusion,
data,
dataloader,
prior,
posterior,
batch_size,
microbatch,
lr,
ema_rate,
log_interval,
save_interval,
resume_checkpoint,
use_fp16=False,
fp16_scale_growth=1e-3,
schedule_sampler=None,
weight_decay=0.0,
lr_anneal_steps=0,
):

def run_loop(self):
    i = 0
    totseg = 0
    totcls = 0
    totrec=0
    data_iter = iter(self.dataloader)
    while (
        not self.lr_anneal_steps
        or self.step + self.resume_step < self.lr_anneal_steps
    ):