SantMan
April 7, 2019, 5:21am
13
mhubii:
struct NetImpl : torch::nn::Module { // replaced Net by NetImpl NetImpl() // replaced Net by NetImpl : conv1(torch::nn::Conv2dOptions(1, 10, /kernel_size= /5)), conv2(torch::nn::Conv2dOptions(10, 20, /kernel_size= /5)), fc1(320, 50), fc2(50, 10) { register_module(“conv1”, conv1); register_module(“conv2”, conv2); register_module(“conv2_drop”, conv2_drop); register_module(“fc1”, fc1); register_module(“fc2”, fc2); } torch::Tensor forward(torch::Tensor x) { x = torch::relu(torch::max_pool2d(conv1->forward(x), 2)); x = torch::relu( torch::max_pool2d(conv2_drop->forward(conv2->forward(x)), 2)); x = x.view({-1, 320}); x = torch::relu(fc1->forward(x)); x = torch::dropout(x, /p= /0.5, /training= /is_training()); x = fc2->forward(x); return torch::log_softmax(x, /dim= /1); } torch::nn::Conv2d conv1; torch::nn::Conv2d conv2; torch::nn::FeatureDropout conv2_drop; torch::nn::Linear fc1; torch::nn::Linear fc2; }; TORCH_MODULE(Net);
@mhubii ,
I’m trying to implement a simplified version fhe code you posted above. I’m facing some issues can you please have a look at this post and let me know what is that I’m missing?