Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- trn_df = trn_df.iloc[k:] #trn_df - исходный большой .csv файл, т.е оригинальный датасет; k - шаги(сколько уже обучился), 200_000, 400_000;
- trn_dataset = MyDataset(trn_df)
- dataloaders['train'] = DataLoader(
- dataset=trn_dataset,
- shuffle=False,
- batch_size=batch_size,
- pin_memory=True,
- num_workers=2,
- drop_last=True,
- collate_fn=collator)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement