| 12 | |
| 13 | |
| 14 | class PSPModuleImpl : public torch::nn::Module { |
| 15 | public: |
| 16 | PSPModuleImpl(int in_channels, std::vector<int> sizes, bool use_bathcnorm = true); |
| 17 | torch::Tensor forward(torch::Tensor x); |
| 18 | private: |
| 19 | torch::nn::ModuleList blocks; |
| 20 | }; TORCH_MODULE(PSPModule); |
| 21 | |
| 22 | class PSPDecoderImpl : public torch::nn::Module { |
| 23 | public: |
nothing calls this directly
no outgoing calls
no test coverage detected