1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#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 utils
} // namespace autograd
} // namespace torch
|
