INNER CODE UNIT · Python
global_iter
FoundationVision/UniTok · main.py:110
global_iter = epoch * num_iters + cur_iter
disc_global_iter = global_iter - disc_start_iter
unitok_lr_stats = unitok_scheduler.step(global_iter)
disc_lr_stats = disc_scheduler.step(disc_global_iter) if disc_global_iter >= 0 else [0]
unitok_lr_stats = list(set(unitok_lr_stats))
disc_lr_stats = list(set(disc_lr_stats))
stepping = (global_iter + 1) % args.grad_accu == 0
warmup_disc_schedule = 0 if disc_global_iter < 0 else min(1.0, disc_global_iter / disc_warmup_iter)
fade_blur_schedule = 0 if disc_global_iter < 0 else min(1.0, disc_global_iter / (disc_warmup_iter * 2))
fade_blur_schedule = 1 - fade_blur_schedule
trainer.train_step(
img=imgs,
text=texts,
global_iter=global_iter,
stepping=stepping,