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.
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.