I am trying to extract features.
extracted_features = []
main_extracted_features = []
for batch in range(100):
print(“Batch Number”,batch)
data_for = new_data[batch*100:(batch+1)*100]
for i in range(len(data_for)):
img = torch.tensor(data_for[i]/255).float()
ft_extract = feature_extractor(img.to(device))
extracted_features.append(ft_extract)
main_extracted_features.append(extracted_features)
print(np.array(extracted_features).shape)
print(np.array(main_extracted_features).shape)
and then crashes
Screenshot 2021-01-09 at 1.16.28 PM996×524 73.6 KB