如何用其他代码导入CNN模型？

问题描述

这是我的“模型”，出于理解目的，我在此处添加了一部分。我希望将此模型导入另一段名为“ trainer”的代码中。

import torch.nn as nn
import torch

class Signet(nn.Module):
    def __init__(self,num_users=10):
        super(Signet,self).__init__()
        self.num_classes = num_users
        self.forward_util_layer = {}

这是另一段代码。我要像上面提到的那样在这里导入“模型”。

import model

class Trainer:
    def __init__(self,trainData,testData,num_users=10):
        self.trainData = trainData
        self.testData = testData

导入它的方式是什么。在哪里保存我的model.ipynb，以便我的trainer.ipynb可以导入它。还是有什么办法可以直接在我的trainer.ipynb代码中调用它？

解决方法

将模型代码移动到<xsl:template name="priceSumSegmented"> <xsl:param name="productList"/> <xsl:param name="segmentLength" select="5"/> <xsl:choose> <xsl:when test="count($productList) > 0"> <xsl:variable name="recursive_result1"> <xsl:call-template name="priceSum"> <xsl:with-param name="productList" select="$productList[position() <= $segmentLength]"/> </xsl:call-template> </xsl:variable> <xsl:variable name="recursive_result2"> <xsl:call-template name="priceSumSegmented"> <xsl:with-param name="productList" select="$productList[position() > $segmentLength]"/> </xsl:call-template> </xsl:variable> <xsl:value-of select="$recursive_result1 + $recursive_result2"/> </xsl:when> <xsl:otherwise><xsl:value-of select="0"/></xsl:otherwise> </xsl:choose> </xsl:template> 文件中。将该文件与model.py

放在同一目录中

现在，要将模型加载到ipynb文件中，请按照下列步骤操作。

trainer.ipynb