def find_ema_checkpoint()

in improved_diffusion/train_util.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

def find_ema_checkpoint(main_checkpoint, step, rate):
    if main_checkpoint is None:
        return None
    filename = f"ema_{rate}_{(step):06d}.pt"
    path = bf.join(bf.dirname(main_checkpoint), filename)
    if bf.exists(path):
        return path
    return None