def get_training_cost(
    model_path: str,
    dataset_path: str,
    train_batch_size: int,
    run_stats: dict,
    gpu_cost_per_hr: float = 0.35,  # GCP cost for Tesla T4
) -> float:
    """
    Return total cost to train model using GCP compute resource
    """
    total_time_s = int(run_stats["hyperopt_results"]["time_total_s"])
    total_time_hr = total_time_s / 3600
    return float(total_time_hr * gpu_cost_per_hr)