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
Hi,I'm getting this error TypeError: Can't broadcast (128, 200704) -> (128, 1024, 14, 14) when running extract_features.py. The error is at line 133 output_h5["features"][counter : counter + len(batch)] = output_features.cpu().numpy()
The text was updated successfully, but these errors were encountered:
Not sure if you still need this (hopefully this could be helpful for others), but I believe that you can simply reshape the output_features to (128, 1024, 14, 14).
EX: After the error line, insert a "output_features = np.reshape(output_features, (128, 1024, 14, 14))" seems to work for me. Let me know if this runs into any other issues.
Hi,I'm getting this error
TypeError: Can't broadcast (128, 200704) -> (128, 1024, 14, 14)
when running extract_features.py. The error is at line 133output_h5["features"][counter : counter + len(batch)] = output_features.cpu().numpy()
The text was updated successfully, but these errors were encountered: