Number of model parameters about transenet HOT 3 CLOSED

Thanks for your test : )

from transenet.

Hi, I just used thop library to compute the number of TransENet and I found that it is almost the same as yours, where the specific number is 37458883. The following code is used to perform the parameter calculations.

  from thop import profile
  input = torch.randn(1, 3, 48, 48).to(device)
  macs, params = profile(sr_model, inputs=(input, ))
  print('Params: ' + str(params))
  print('MACs: ' + str(macs))

from transenet.

Oh! Thanks for the quick answer, I tested the mod using your code and got the same results as you.
Also, I tested the same parameters as I did with your previous work on HSENet and the results were as follows (Figure 1 for HSENet and Figure 2 for TransENet).

Thanks this time！

from transenet.