Advertisement
Pastehsjsjs

Untitled

Apr 2nd, 2023
578
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.45 KB | Source Code | 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)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement