Commit e00a4560 authored by Bucky Lee's avatar Bucky Lee
Browse files

backend update

parent 677beff1
Loading
Loading
Loading
Loading
+3 −1
Original line number Diff line number Diff line
@@ -4,7 +4,9 @@
    <option name="enabled" value="true" />
  </component>
  <component name="NewModuleRootManager">
    <content url="file://$MODULE_DIR$" />
    <content url="file://$MODULE_DIR$">
      <sourceFolder url="file://$MODULE_DIR$/models" isTestSource="false" />
    </content>
    <orderEntry type="inheritedJdk" />
    <orderEntry type="sourceFolder" forTests="false" />
  </component>
+1.93 KiB

File added.

No diff preview for this file type.

+163 B

File added.

No diff preview for this file type.

+159 B

File added.

No diff preview for this file type.

+3 −3
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ See our template model class 'template_model.py' for more details.
"""

import importlib
from models.base_model import BaseModel
from Adaptation_model.base_model import BaseModel


def find_model_using_name(model_name):
@@ -58,8 +58,8 @@ def create_model(opt):
    This is the main interface between this package and 'train.py'/'test.py'

    Example:
        >>> from models import create_model
        >>> model = create_model(opt)
        #>>> from models import create_model
        #>>> model = create_model(opt)
    """
    model = find_model_using_name(opt.model)
    instance = model(opt)
Loading