The epochs for fb1k5237 and wn18nn

[quqxui]

Hi, thanks for your great work. I want to know how many epochs you need to get the reported results of fb1k5237 and wn18nn on your paper?

As you mentioned that training on a single 3090 GPU takes 2-3 days, it seems unlikely that the default 500 epochs in the main.py file are necessary. Perhaps the number of epochs could be reduced to improve training time without sacrificing the model's performance.