10 lines
260 B
Python
10 lines
260 B
Python
from .models import Backbone_nFC, Backbone_nFC_Id
|
|
|
|
|
|
def get_model(model_name, num_label, use_id=False, num_id=None):
|
|
if not use_id:
|
|
return Backbone_nFC(num_label, model_name)
|
|
else:
|
|
return Backbone_nFC_Id(num_label, num_id, model_name)
|
|
|