dgs.models.engine.dgs_engine.DGSEngine.get_target

DGSEngine.get_target(ds: State) → torch.Tensor

Get the target data.

For the similarity engine, the target data consists of the dataset-unique class-id. The get_target() function will be called twice, once for the current time-step and once for the previous.