Open kdoh0914 opened 1 year ago
Hi. Thanks for your concern. You can calculate model size by this code:
for _, p in model.named_parameters():
paras += p.numel()
As for train epoch, you can refer to main_informer.py
.
Inference speed is different in different operating environments, you can do a simple test using time.time()
.
Hi. Can you share your
Thanks a lot!