- 博客(2)
- 收藏
- 关注
RSS订阅原创 ResNet神经网络Pytorch版源码解读
以下代码定义了ResNet网络的名称，各个名称对应网络的预训练参数的下载地址，以及经常用到的Conv3*3卷积层。import torch.nn as nnimport mathimport torch.utils.model_zoo as model_zoo__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101', 'resnet152']model_urls = { 'resnet18
2020-09-14 19:59:43
939
原创 tensorflow2中使用ragged.constant()将非矩形列表转为tensor类型
在进行数据标签的处理时，标签数据是一个非矩形列表，使用tf.constant()无法处理，使用tf.ragged.constant()可将其转换为tensorflow可以直接处理的tensor类型数据。train_labels = [[1, 3], [1,5,4], [2,5,6,3], [2,3,4,8,2], [9]]train_labels = tf.ragged
2020-06-23 10:42:35
1109
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人