TODO revisit whitening object to be compatible with torch.transforms composition.
Currently the whitener init function takes data as an argument. In the future, we should consider removing this from the init, instead of including only arguments to be passed which are parameters of the whitening. The mean and other member variables will instead be computed and initialized after object init.
TODO revisit whitening object to be compatible with torch.transforms composition. Currently the whitener init function takes
dataas an argument. In the future, we should consider removing this from the init, instead of including only arguments to be passed which are parameters of the whitening. The mean and other member variables will instead be computed and initialized after object init.@ligeralde @alvinzz