UPS

class daart.callbacks.UPS(prob_threshold=0.95, variance_threshold=0.05, epoch_start=10)

Bases: BaseCallback

Implement uncertainty-aware pseudo-labels algorithm.

See details in: https://arxiv.org/pdf/2101.06329.pdf

Methods Summary

on_epoch_end(data_generator, model, trainer, ...)

Methods Documentation

on_epoch_end(data_generator, model, trainer, **kwargs)