ngclearn.utils.masks package

Submodules

ngclearn.utils.masks.multiblock2d module

class ngclearn.utils.masks.multiblock2d.MaskCollator(cfgs_mask, crop_size=(224, 224), patch_size=(16, 16))

Bases: object

A mechanism for generating/creating patch masks, generally for self-supervised learning.

Parameters:

• cfgs_mask – configuration masks to apply

• crop_size – dimensions of crop

• patch_size – dimensions of patches to create

step()

Steps this generator forward one step.

Returns:

next set of collated encoder masks, next set of predictor masks

Module contents