In test_demo.py, line89-line90
"""
print(f"\tTotal # of model parameters : {num_params / (1000**2) :.3f} M")
print(f"\tTotal FLOPs of the model : {flops.total() / (1000**4) :.3f} T")
"""
Should the divisor be 1024 while converting?
Hi,
We think it's a minor issue and believe 1000 is correct. (The unit comes from the SI prefix, not the byte)
Even if it's not, we maintain the code for a fair comparison. Please follow the original code we provided.
Thx.
Posted by: THQiu @ March 18, 2025, 6:51 a.m.