(net, num_class=13, temp=0.05, top=False, norm=True)
| 6 | |
| 7 | |
| 8 | def get_model_mme(net, num_class=13, temp=0.05, top=False, norm=True): |
| 9 | dim = 2048 |
| 10 | if "resnet" in net: |
| 11 | model_g = ResBase(net, top=top) |
| 12 | if "resnet18" in net: |
| 13 | dim = 512 |
| 14 | if net == "resnet34": |
| 15 | dim = 512 |
| 16 | elif "vgg" in net: |
| 17 | model_g = VGGBase(option=net, pret=True, top=top) |
| 18 | dim = 4096 |
| 19 | if top: |
| 20 | dim = 1000 |
| 21 | print("selected network %s"%net) |
| 22 | return model_g, dim |
| 23 | |
| 24 | def log_set(kwargs): |
| 25 | source_data = kwargs["source_data"] |
no test coverage detected