zhouhaoyi / Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021.
Apache License 2.0
5.4k stars 1.12k forks source link

Model size and performance #533

Open kdoh0914 opened 1 year ago

kdoh0914 commented 1 year ago

Hi. Can you share your

Thanks a lot!

MountVoom commented 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().