batch_size to 1

laitassou · Aug 24, 2020 · ee0ba91 · ee0ba91
1 parent 32fbbab
commit ee0ba91
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/model_affine/affine/affine.onnx b/model_affine/affine/affine.onnx
diff --git a/model_affine/affine/saved_model.pb b/model_affine/affine/saved_model.pb
diff --git a/model_affine/affine/variables/variables.data-00000-of-00001 b/model_affine/affine/variables/variables.data-00000-of-00001
diff --git a/model_affine/affine/variables/variables.index b/model_affine/affine/variables/variables.index
diff --git a/model_affine/model_affine.py b/model_affine/model_affine.py
@@ -12,8 +12,8 @@
 y = np.array([-2,-5, -7, -12 ,-15, -5, -12,-10,-5],np.float32)
 
 
-BATCH_SIZE  = 4
-num_epochs = 1000
+BATCH_SIZE  = 1
+num_epochs = 200
 
 dataset = tf.data.Dataset.from_tensor_slices(( x , y ))
 
@@ -29,7 +29,7 @@
 
 model.compile(optimizer='sgd', loss='mse')
 
-model.fit(dataset, epochs=num_epochs, steps_per_epoch=tf.data.experimental.cardinality(dataset).numpy() // BATCH_SIZE -1, verbose=0)
+model.fit(dataset, epochs=num_epochs, steps_per_epoch=tf.data.experimental.cardinality(dataset).numpy() ,  batch_size = 1, verbose=0)
 
 print(model.layers[0].get_weights())