search

CalculatedContent / WeightWatcher

The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
Apache License 2.0
1.47k stars 125 forks source link

WeightWatcher.ipynb 'numpy.ndarray' object has no attribute 'extend' #306

Closed Hadar933 closed 8 months ago

Hadar933 commented 8 months ago

the jupyter notebook WeightWatcher.ipynb craches due to an Attribute error numpy.ndarray object has no attribute extend, specifically during the execution of watcher = ww.WeightWatcher(model=model). This is for

import tensorflow as tf
from tensorflow.keras.models import load_model
from tensorflow.keras.applications import vgg16

kmodel = vgg16.VGG16
model = kmodel(weights='imagenet')

tested locally and on google colab

charlesmartin14 commented 8 months ago

This has been fixed in version 0.7.5.1 If you find any additional problems like this, please file a bug and I will fix it ASAP