"...api/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "46af98267d668ae9fa61ac8a5e1e4e5c229f58d3"
Commit ceec4566 authored by Davis King's avatar Davis King
Browse files

Added some checks that the user produces PSI vectors that are the proper

dimension.
parent afa4fe0c
...@@ -75,6 +75,12 @@ namespace dlib ...@@ -75,6 +75,12 @@ namespace dlib
psi = true_psi; psi = true_psi;
else else
prob->get_truth_joint_feature_vector(sample_idx, psi); prob->get_truth_joint_feature_vector(sample_idx, psi);
if (is_matrix<feature_vector_type>::value)
{
DLIB_CASSERT(psi.size() == (unsigned long)prob->get_num_dimensions(),
"The dimensionality of your PSI vector doesn't match get_num_dimensions()");
}
} }
void separation_oracle_cached ( void separation_oracle_cached (
...@@ -128,6 +134,11 @@ namespace dlib ...@@ -128,6 +134,11 @@ namespace dlib
prob->separation_oracle(sample_idx, current_solution, out_loss, out_psi); prob->separation_oracle(sample_idx, current_solution, out_loss, out_psi);
if (is_matrix<feature_vector_type>::value)
{
DLIB_CASSERT(out_psi.size() == (unsigned long)prob->get_num_dimensions(),
"The dimensionality of your PSI vector doesn't match get_num_dimensions()");
}
if (!cache_enabled) if (!cache_enabled)
return; return;
......
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