<reponame>DarshanDeshpande/research-paper-implementations
hyperparameters = {
    "IMG_HEIGHT": 256,
    "IMG_WIDTH": 256,
    "PROJECTION_DIM": 512,
    "PATCH_SIZE": 8,
    "EMB_DIM": 512,
    "NUM_DECODER_LAYERS": 3,
    "NUM_HEADS": 4,
    "DECODER_OUTPUT_DIM": 512,
}

loss_params = {"lambda_c": 10, "lambda_s": 7, "lambda_id1": 50, "lambda_id2": 1}

hyperparameters = {
    "IMG_HEIGHT": 256,
    "IMG_WIDTH": 256,
    "PROJECTION_DIM": 512,
    "PATCH_SIZE": 8,
    "EMB_DIM": 512,
    "NUM_DECODER_LAYERS": 3,
    "NUM_HEADS": 4,
    "DECODER_OUTPUT_DIM": 512,
}

loss_params = {"lambda_c": 10, "lambda_s": 7, "lambda_id1": 50, "lambda_id2": 1}