JiuFang

摘要： 说明：返回每一层的参数的名称和参数内容（权重和偏置） 作用：一般来说，类中的成员都是私有的，可以通过这种方式获得模型中的参数 例如： import torch import torch.nn as nn class TestModel(nn.Module): def __init__(self): 阅读全文

摘要： 作用： model.named_children() #返回一个迭代器，该迭代器能返回模块的名称以及模块本身 model.named_modules() #返回一个迭代器，该迭代器返回网络中所有模块的名字和模块本身 ''' 1.会返回组成子模块的模块 2.返回的第一个值是整个模型 ''' 测试： i 阅读全文

摘要： 用途：用于指定将数据或者模型放在GPU/CPU #1.通常用法 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") data = data.to(device) model = model.to(device) 阅读全文

摘要： 相同： 两者都是导入numpy库的函数，变量，对象等 不同： 使用import numpy导入时，若要使用库中的相关函数，对象等，需要在前面加上'库名.' 使用from numpy import *导入时，无需添加库名，直接使用相关的函数即可 例子： import numpy #调用array（） 阅读全文