无法从Google Collab中的另一个python文件导入类

问题描述

我正在尝试将另一个python文件中的类导入到Google colab笔记本中。

类定义为-

class model_bn(nn.Module):
    def __init__(self,feature_size=512,classes_num=75):

        super(model_bn,self).__init__() 

        self.features_1 = nn.Sequential(*list(VGG('VGG16').features.children())[:34])
        self.features_2 = nn.Sequential(*list(VGG('VGG16').features.children())[34:])

        self.max = nn.MaxPool2d(kernel_size=2,stride=2)
        #self.features_list = features_list
        self.num_ftrs = 225*7*7 #num_classes * num_channels
        self.classifier = nn.Sequential(
            nn.BatchNorm1d(self.num_ftrs),#nn.Dropout(0.5),nn.Linear(self.num_ftrs,feature_size),nn.BatchNorm1d(feature_size),nn.ELU(inplace=True),nn.Linear(feature_size,classes_num),)

我已将此类的文件加载到正确的工作区中,可以导入正常的函数,但是当我尝试导入类时,它说-cannot import model_bn

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)