Prince/models/15March_Dlight/dfl_sae_state.json

90 lines
2.0 KiB
JSON

{
"name": "dfl_sae",
"sessions": {
"1": {
"timestamp": 1647369535.5817182,
"no_logs": false,
"loss_names": [
"total",
"face_a",
"mask_a",
"face_b",
"mask_b"
],
"batchsize": 16,
"iterations": 60,
"config": {
"learning_rate": 5e-05,
"epsilon_exponent": -7,
"allow_growth": false,
"nan_protection": true,
"convert_batchsize": 16,
"eye_multiplier": 10,
"mouth_multiplier": 20,
"clipnorm": true
}
},
"2": {
"timestamp": 1647462019.6170704,
"no_logs": false,
"loss_names": [
"total",
"face_a",
"mask_a",
"face_b",
"mask_b"
],
"batchsize": 8,
"iterations": 8340,
"config": {
"learning_rate": 5e-05,
"epsilon_exponent": -7,
"allow_growth": false,
"nan_protection": true,
"convert_batchsize": 16,
"eye_multiplier": 10,
"mouth_multiplier": 20,
"clipnorm": true
}
}
},
"lowest_avg_loss": {
"a": 0.0803792018443346,
"b": 0.038223030604422094
},
"iterations": 8400,
"config": {
"centering": "face",
"coverage": 87.5,
"optimizer": "adam",
"learning_rate": 5e-05,
"epsilon_exponent": -7,
"allow_growth": false,
"mixed_precision": false,
"nan_protection": true,
"convert_batchsize": 16,
"loss_function": "ssim",
"mask_loss_function": "mse",
"l2_reg_term": 100,
"eye_multiplier": 10,
"mouth_multiplier": 20,
"penalized_mask_loss": true,
"mask_type": "extended",
"mask_blur_kernel": 0,
"mask_threshold": 4,
"learn_mask": true,
"input_size": 128,
"clipnorm": true,
"architecture": "df",
"autoencoder_dims": 0,
"encoder_dims": 42,
"decoder_dims": 21,
"multiscale_decoder": false,
"freeze_layers": [
"encoder_df"
],
"load_layers": [
"encoder_df"
]
}
}