↓ 4 callersFunctionload_2MNIST(
n_digits=10,
dataset_dimensions={"train": 42000, "val": 12000, "test": 6000},
c_sup=1,
which
expressive/experiments/rsbench/datasets/utils/mnist_creation.py:220
↓ 2 callersMethodKANDsat_agg_loss(self, shapes, colors, conc_preds, labels, args, b_idx)
expressive/experiments/rsbench/utils/kand_ltn_loss.py:32
↓ 2 callersMethod__init__(self, x_shape=(3, 64, 64), z_size=6, z_multiplier=1)
expressive/experiments/rsbench/backbones/disent_encoder_decoder.py:17