1. Learn
  2. /
  3. Kurser
  4. /
  5. Skalowalne modele AI z PyTorch Lightning

Connected

övning

Tworzenie DataLoadera dla danych treningowych

Skoro podzieliliśmy już zbiór danych, musimy zdefiniować data loader, który będzie dostarczał partie danych podczas treningu. DataLoader sprawnie wczytuje dane do pamięci i umożliwia ich losowanie, co poprawia generalizację modelu. W tym ćwiczeniu uzupełnisz metodę train_dataloader.

Instruktioner

100 XP
  • Zaimportuj DataLoader.
  • Zwróć DataLoader, który wczytuje self.train_data z włączonym losowaniem kolejności danych, aby poprawić generalizację.