I want to fit this code with my personal dataset. Not the IMDB ones. The load_data_imdb() returns 3 parameters.
To fit my dataset in, I’ve to set the 3 parameters right? But what value should I put?
import torch from torch import nn from d2l import torch as d2l batch_size = 64 train_iter, test_iter, vocab = d2l.load_data_imdb(batch_size)