CalvinBackup/neuralchen-SimSwap
1
0
Fork 0
Code Issues 288 Pull Requests 16 Actions Packages Projects Releases 2 Wiki Activity

Update m1_tf_test.py

Browse Source
This commit is contained in:
Smiril
2022-07-11 09:53:45 +02:00
committed by GitHub
parent 84ab818563
commit 8daacea01f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
m1_tf_test.py
+2 -2
View File
@@ -26,7 +26,7 @@ class ComputeSum(Layer):
self.total.assign_add(tf.reduce_sum(inputs, axis=0)) self.total.assign_add(tf.reduce_sum(inputs, axis=0))
return self.total return self.total
def Build_Tree(self, data, depth = 0): def call(self, data, depth = 0):
n = len(data) n = len(data)
return n return n
@@ -92,7 +92,7 @@ model.compile(optimizer='adam',
loss=loss_fn, loss=loss_fn,
metrics=['accuracy']) metrics=['accuracy'])
these = model.fit(x_train, y_train, epochs=10) these = model.fit(x_train, y_train, epochs=10)
that = ComputeSum.Build_Tree(these) that = ComputeSum(len(these))
those = that(these) those = that(these)
outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')(those) outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')(those)
model = ComputeSumModel((tf.shape(these)[1],)) model = ComputeSumModel((tf.shape(these)[1],))