PytorchCNNModules

Pytorch CNN Module collection


Keywords
pytorch, cnn, module, inception, block
License
MIT
Install
pip install PytorchCNNModules==0.0.1

Documentation

PyTorch CNN Modules

Introduction

Pytorch CNN Modules is a collection of CNN modules like the Inception Module. These modules can be used in a single line. There is no need to research or implement anything.

Features

All modules can be easily changed to residual mode and dense mode. All modules can add SE module.

Requirement

torch 1.7.0 or above

Install

TODO

Usage

create bottleneck module

from PytorchCNNModules import *
module = BottleneckResidual(in_feature, out_feature).to_residual()

output = module(input)

License

Pytorch CNN Modules is provided as open source under the MIT License, see LICENSE.