Update vgg16.py

idoamihai · Jan 12, 2018 · e6edd0c · e6edd0c
1 parent e081626
commit e6edd0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/vgg16.py b/nbs/vgg16.py
@@ -94,7 +94,7 @@ def ft(self, num):
  self.compile()
 
  def finetune(self, batches):
- self.ft(batches.num_classes) # Keras2
+ self.ft(batches.num_class) # Keras2.0 (will become .num_classes with Keras 2.1)
  classes = list(iter(batches.class_indices))
  for c in batches.class_indices:
  classes[batches.class_indices[c]] = c