摘要: hub模块中有前人已经训练好的模型参数,可以根据情况调用 # hub模块:调用 # torch.hub.list('pytorch/vision:v0.10.0') # 使用前人已有的模型库 # https://github.com/pytorch/hub/blob/master/pytorch_v 阅读全文
posted @ 2024-10-05 14:49 星空28 阅读(10) 评论(0) 推荐(0) 编辑
摘要: 简单线性回归 y = 2*x + 1 import numpy as np import torch import torch.nn as nn class LinearRegressionModel(nn.Module): def __init__(self, input_dim, output_ 阅读全文
posted @ 2024-10-05 13:10 星空28 阅读(18) 评论(0) 推荐(0) 编辑
摘要: 解题思路 二维列表思路 def bag_value(ls, vol): """ :param ls: 水果体积及价值列表,列表长度表示水果种类,ls[i][0]表示体积,ls[i][1]表示价值, :param vol:背包体积 ls = [[3, 1], [4, 5], [5, 6], [7, 9 阅读全文
posted @ 2024-10-05 04:49 星空28 阅读(5) 评论(0) 推荐(0) 编辑