Advertisement
Guest User

Untitled

a guest
Mar 29th, 2020
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.58 KB | None | 0 0
  1. def combineDataloaders2(loader1,loader2):
  2. combineddata = []
  3. combinedtarget = []
  4.  
  5. for batch in loader1:
  6. for img in batch[0]:
  7. combineddata.append(img)
  8. for label in batch[1]:
  9. combinedtarget.append(label)
  10.  
  11. for batch in loader2:
  12. for img in batch[0]:
  13. combineddata.append(img)
  14. for label in batch[1]:
  15. combinedtarget.append(label)
  16.  
  17.  
  18. print(combineddata[0].shape)
  19. combinedDataset = MyCombinedDataset(combineddata, combinedtarget)
  20. dataloader = DataLoader(combinedDataset, batch_size=batch_size)
  21. return dataloader
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement