Learning Day 22: What is nn.Module in Pytorch
Benefits of using nn.Module
- nn.Module can be used as the foundation to be inherited by model class
import torch
import torch.nn as nn
class…
All Stories published by dejunhuang on May 07, 2021
About
dejunhuang
For the purpose of systematic learning, I will write down my daily learning notes on data science and machine learning
More informationTags
Editors
