Advertisement
Pastehsjsjs

Untitled

Apr 2nd, 2023
64
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.45 KB | None | 0 0
  1. trn_df = trn_df.iloc[k:] #trn_df - исходный большой .csv файл, т.е оригинальный датасет; k - шаги(сколько уже обучился), 200_000, 400_000;
  2. trn_dataset = MyDataset(trn_df)
  3.  
  4. dataloaders['train'] = DataLoader(
  5. dataset=trn_dataset,
  6. shuffle=False,
  7. batch_size=batch_size,
  8. pin_memory=True,
  9. num_workers=2,
  10. drop_last=True,
  11. collate_fn=collator)
  12.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement