Skip to content

Commit

Permalink
Code-format
Browse files Browse the repository at this point in the history
  • Loading branch information
iarspider committed Nov 29, 2023
1 parent 90c3620 commit db19114
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions PhysicsTools/PythonAnalysis/test/time_serie_prediction.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,10 @@
#include <iostream>

struct Net : torch::nn::Module {
Net(int64_t N, int64_t M)
: linear(register_module("linear", torch::nn::Linear(N, M))) {
Net(int64_t N, int64_t M) : linear(register_module("linear", torch::nn::Linear(N, M))) {
another_bias = register_parameter("b", torch::randn(M));
}
torch::Tensor forward(torch::Tensor input) {
return linear(input) + another_bias;
}
torch::Tensor forward(torch::Tensor input) { return linear(input) + another_bias; }
torch::nn::Linear linear;
torch::Tensor another_bias;
};
Expand Down

0 comments on commit db19114

Please sign in to comment.