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

various additions for M1 GPU #299

Closed
Smiril wants to merge 80 commits from main into main
pull from: main
merge into: :main
Conversation 0 Commits 80 Files Changed 12
+1 -1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 241ccdd253 - Show all commits
m1_tf_test.py
+1 -1
View File
@@ -89,7 +89,7 @@ model = tf.keras.models.Sequential([
tf.keras.layers.Dropout(0.2)
])
loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.add(LSTM(32, input_shape=(1024, )))
model.add(LSTM(self,32, input_shape=(1024, )))
model.compile(optimizer='adam',
loss=loss_fn,
metrics=['accuracy'])