Commit 792ed396 authored by Marius Laska's avatar Marius Laska

changed optimizer since adamax produces nan values on GPU

parent ddd51972
......@@ -134,7 +134,7 @@ def ae_model(X_cols, params):
encoder_model.compile(
loss=MSE,
optimizer=Adamax(lr=params['lr'])#lr=0.005)
optimizer=Adam(lr=params['lr'])#lr=0.005)
)
return encoder_model, bottleneck_model
......@@ -163,7 +163,7 @@ def cnn_deep_loc_box(ae_base_model_file, y_cols, params):
position_model.compile(
loss=params['losses'],
optimizer=Adamax(lr=params['lr'])#lr=0.005)
optimizer=Adam(lr=params['lr'])#lr=0.005)
)
return position_model
......@@ -42,7 +42,7 @@ global_params:
#hidden_layers: 1
lr: 0.002
batch_size: 32
epochs: 1
epochs: 200
#dropout: 0.7
#regularization_penalty: 0.0
augmentation: 0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment