hi, Can you help me please. I want to use this code for my own images dataset but when I execute the function fit(train_loader, test_loader, model, loss_fn, optimizer, scheduler, n_epochs, cuda, log_interval, metrics=[AccumulatedAccuracyMetric()]), I got the Error :Expected 4-dimensional input for 4-dimensional weight 64 1 10 10, but got 2-dimensional input of size [100, 16384] instead.

[sousoufa]