models.__dict__[args.arch] 实例复现-白红宇

models.__dict__[args.arch] 实例复现

阅读量：799 次

发布时间：2023-02-09

本文共 1238 字，大约阅读时间需要 4 分钟。

在查看moco源码时，我发现了一个复杂的代码段，决定尝试复现它。以下是我理解和复现过程的详细记录。

代码段复现

model = moco.builder.MoCo(    models.__dict__[args.arch],    args.moco_dim,    args.moco_k,    args.moco_m,    args.moco_t,    args.mlp)print(model)

文件结构

项目文件夹：

仅需三个.py文件：test.py、test/init.py、test2.py

test文件夹：

__init__.py：包含以下代码
```
from .test2 import *
```

test2.py：定义多个函数

__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101']def ResNet():    print('ResNet')def resnet18():    print('resnet18')def resnet34():    print('resnet34')def resnet50():    print('resnet50')def resnet101():    print('resnet101')

test.py文件：

import testclass Test:    def __init__(self, model_name='', age='', sex=''):        self.model_name = model_name        self.age = age        self.sex = sex        print('self.model_name', self.model_name)        print('self.age', self.age)        print('self.sex', self.sex)    def __call__(self):        print('调用函数')new_test = Test(test.__dict__['resnet50'], '18', 'boy')print('----------')test.__dict__['resnet50']()print(new_test.model_name)new_test.model_name()