Because the dimension (224x224x3) is too large, I noticed that the ready-made dist dimension provided by the author is 3072 (32x32x3).