Paras Koundal

@xuyxu It is simple. ``` ..... from torch_geometric.loader import DataLoader ....... train_loader = DataLoader(train_dataset, batch_size= batch_s, shuffle=True,drop_last=True) val_loader = DataLoader(val_dataset, batch_size=2,drop_last=True) test_loader = DataLoader(test_dataset, batch_size=2,drop_last=True) ....... ``` I have created...

@xuyxu Here's that ``` test_loader=test_loader File "/cr/data02/koundal/applications/gpu-project/lib/python3.7/site-packages/torchensemble/bagging.py", line 329, in fit self.n_outputs = self._decide_n_outputs(train_loader) File "/cr/data02/koundal/applications/gpu-project/lib/python3.7/site-packages/torchensemble/_base.py", line 267, in _decide_n_outputs _, target = split_data_target(elem, self.device) File "/cr/data02/koundal/applications/gpu-project/lib/python3.7/site-packages/torchensemble/utils/io.py", line 84, in...

@xuyxu Any update on this?