MetricFuncCache

class cubie.outputhandling.summarymetrics.metrics.MetricFuncCache(update_fn: Callable = None, save_fn: Callable = None)[source]

Bases: CUDADispatcherCache

Cache container for compiled metric functions.

update

Callable update device function.

save

Callable save device function.

save_fn: Callable
update_fn: Callable