torch杂记
序号 | 名称 | 备注 |
---|---|---|
1 | torchvision.transforms中的ToTensor和Normalize | torchvision |
2 | torch.nn.init.kaiming_normal_ | torch.nn.init |
2 | torch.nn.parallel.DistributedDataParallel | DistributedDataParallel后台单机多卡训练 |
torch练习
序号 | 名称 | 备注 |
---|---|---|
1 | torch_linear_regression | linear regression |
2 | torch_logistic_regression | logistic regression |
3 | torch_feed_mynet | 自定义network |
4 | 自定义卷积权重 | 自定义卷积权重 |
5 | 计算多分类交叉熵 | 自定义卷积权重 |