Skip to content

Commit 9b0b4ce

Browse files
authored
Update train_net.cpp
1 parent 94ac920 commit 9b0b4ce

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

source/train_net.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@ double acc(torch::Tensor& preds, torch::Tensor& labels){
99
}
1010

1111
int main() {
12-
torch::manual_seed(1);
12+
// Set the seed.
13+
1314
// Create a new Net.
1415
auto net = std::make_shared<Net>();
1516

0 commit comments

Comments
 (0)