You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I think there might be a mistake in the pooling op count. Right now in ops_pooling, the number of operations is independent of the number of input planes. I think that the number returned should be multiplied by the number of input planes.
The text was updated successfully, but these errors were encountered:
@gpleiss , hi geoff I happened to use this code too! I manually computed flops of a model using python script, and only the result of pooling is significantly different from this code. Great to see you pointed out the bugs.
I think there might be a mistake in the pooling op count. Right now in ops_pooling, the number of operations is independent of the number of input planes. I think that the number returned should be multiplied by the number of input planes.
The text was updated successfully, but these errors were encountered: