Hello, thank you for your nice work. There is a problem with one of the examples. The model does not train anymore when changing batch_size in test_loader in shapes.py, i.e. it trains much worse when decrasing the batch_size to 10 or 5 (emphasis: in test_loader not train_loader!) and with test_loader at batch_size=1 the best performance is achieved after only a few epochs and it is at the level of 0.3. Is this to be expected? Is the role of the test dataset somehow different than normally and is supposed to leak information during training? Thanks in advance.
Hello, thank you for your nice work. There is a problem with one of the examples. The model does not train anymore when changing
batch_sizeintest_loaderinshapes.py, i.e. it trains much worse when decrasing thebatch_sizeto10or5(emphasis: intest_loadernottrain_loader!) and withtest_loaderatbatch_size=1the best performance is achieved after only a few epochs and it is at the level of0.3. Is this to be expected? Is the role of the test dataset somehow different than normally and is supposed to leak information during training? Thanks in advance.