Method PSPModuleImpl

src/architectures/PSPNetDecoder.cpp:23–28

PSPModuleImpl::PSPModuleImpl(int in_channels, std::vector<int> _sizes, bool use_bathcnorm) {
	for (auto size : _sizes) {
		blocks->push_back(PSPBlock(in_channels, in_channels / _sizes.size(), size, use_bathcnorm));
	}
	register_module("blocks", blocks);
}

torch::Tensor PSPModuleImpl::forward(torch::Tensor x) {
	std::vector<torch::Tensor> xs;