now yes
This commit is contained in:
@@ -271,9 +271,8 @@ x = Dense(512, activation='relu')(x) # Added another Dense layer
|
||||
x = Dense(256, activation='relu')(x) # Existing Dense layer
|
||||
predictions = Dense(1, activation='sigmoid')(x) # Output layer for binary classification
|
||||
|
||||
with strategy.scope(): # Use all gpus
|
||||
model = Model(inputs=base_model.input, outputs=predictions)
|
||||
model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
|
||||
model = Model(inputs=base_model.input, outputs=predictions)
|
||||
model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
|
||||
|
||||
"""## 4. Data Generators
|
||||
|
||||
|
||||
Reference in New Issue
Block a user