#pragma once
|
|
#include <sstream>
|
|
namespace torch { namespace autograd { namespace utils {
|
|
inline std::string requires_grad_leaf_error(bool requires_grad) {
|
std::ostringstream oss;
|
oss << "you can only change requires_grad flags of leaf variables.";
|
if (requires_grad == false) {
|
oss << " If you want to use a computed variable in a subgraph "
|
"that doesn't require differentiation use "
|
"var_no_grad = var.detach().";
|
}
|
return oss.str();
|
}
|
|
}}} // namespace torch::autograd::utils
|
