four_simEps_model.py

#Import the required libraries
import os
import torch
import torch.nn as nn
import functools
from utilities.m_net_factory import net_factory
from utilities.model_initialization import kaiming_normal_init_weight, xavier_normal_init_weight, xavier_uniform_init_weight, sparse_init_weight



# model_1 = nn.DataParallel(kaiming_normal_init_weight(net_factory(net_type='Unet', in_chns=3, class_num=4)))
# model_2 = nn.DataParallel(xavier_normal_init_weight(net_factory(net_type='Unet', in_chns=3, class_num=4)))

model1 = net_factory(net_type='unet_f', in_chns=3, class_num=4)  
model2 = net_factory(net_type='unet_f', in_chns=3, class_num=4)  
model3 = net_factory(net_type='unet_f', in_chns=3, class_num=4)  
model4 = net_factory(net_type='unet_f', in_chns=3, class_num=4)  
# model2 = xavier_normal_init_weight(model2)
# model3 = sparse_init_weight(model3)
# model4 = kaiming_normal_init_weight(model4)
 

model1 = nn.DataParallel(model1)
model2 = nn.DataParallel(model2)
model3 = nn.DataParallel(model3)
model4 = nn.DataParallel(model4)
# print(model1)