Recent posts

nn.Module: Network Construction

BackgroundTo define our network, we should succeed class nn.Module and implement the function forward. We put all the layers we want in the function __init__() and define how layers connect in function