(filename, batch_size, train=True, target="logits")
| 55 | |
| 56 | |
| 57 | def dataloader_gen2(filename, batch_size, train=True, target="logits"): |
| 58 | dataset = data_loader.heter_data2(filename, target=target) |
| 59 | dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=train, num_workers=2, |
| 60 | drop_last=True) |
| 61 | return dataloader |
| 62 | |
| 63 | |
| 64 | def load_pickle(filename): |
nothing calls this directly
no outgoing calls
no test coverage detected
searching dependent graphs…