for data in loader: train = data['input'][:,:-2700] train = train.to(DEVICE) target = data['input'][:,-2700:] target = target.to(DEVICE) model.set_model(data['input_num']) output = model(data)
Every time I input data, I want to call the set_model method for each input.
However, Error occurs because input_num, which is a single value, changes to batch_size.
Is there a good way?
For example, I set the batch_size to 8, and I want to call each of the eight methods in one batch.