We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4e83c96 commit 65a6bdeCopy full SHA for 65a6bde
metatomic-torch/src/outputs.cpp
@@ -195,7 +195,7 @@ static void validate_gradient(
195
if (gradient->samples()->names() != expected_samples_names) {
196
C10_THROW_ERROR(ValueError,
197
"invalid samples for '" + name + "' output '" + parameter + "' gradients: "
198
- "expected the names to be ['sample'], got " +
+ "expected the names to be " + join_names(expected_samples_names) + ", got " +
199
join_names(gradient->samples()->names())
200
);
201
}
0 commit comments