INNER CODE UNIT · Python
eval_interval
FoundationVision/UniTok · main.py:101
eval_interval = int(num_iters // args.eval_per_epoch)
for cur_iter, sample in metric_logger.monitor_enumerate(dataloader, start_iter, num_iters, print_freq=100):
profile_logger.log(cur_iter)
imgs, texts = sample
imgs = imgs.to(args.device, non_blocking=True)
texts = texts.to(args.device, non_blocking=True)
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