Traceback (most recent call last):


 "/home/user/miniconda/envs/py36/bin/stylegan2_pytorch", line 53, in <module>

fire.Fire(train_from_folder)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/fire/core.py", line 138, in Fire

component_trace = _Fire(component, args, parsed_flag_args, context, name)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/fire/core.py", line 471, in _Fire

target=component.__name__)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/fire/core.py", line 675, in _CallAndUpdateTrace

component = fn(*varargs, **kwargs)


 "/home/user/miniconda/envs/py36/bin/stylegan2_pytorch", line 48, in train_from_folder

model.train()


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/stylegan2_pytorch/stylegan2_pytorch.py", line 443, in train

real_output = self.GAN.D(image_batch)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/stylegan2_pytorch/stylegan2_pytorch.py", line 324, in forward

x = self.blocks(x)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/container.py", line 100, in forward

input = module(input)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/stylegan2_pytorch/stylegan2_pytorch.py", line 254, in forward

res = self.conv_res(x)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__

result = self.forward(*input, **kwargs)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 345, in forward

return self.conv2d_forward(input, self.weight)


 "/home/user/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 342, in conv2d_forward

self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size 16 3 1 1, expected input[3, 1, 128, 128] to have 3 channels, but got 1 channels instead

Traceback (most recent call last):


 "multiprocessing\process.py", line 297, in _bootstrap


 "multiprocessing\process.py", line 99, in run


 "paddlexui\pms\model_tasks\tasks.py", line 68, in _call_paddlex_train


 "paddlexui\pms\model_tasks\train\classification.py", line 134, in train


 "site-packages\paddlex\cv\models\classifier.py", line 195, in train


 "site-packages\paddlex\cv\models\base.py", line 262, in net_initialize


 "site-packages\paddlex\utils\utils.py", line 292, in load_pretrain_weights


 "site-packages\paddle\fluid\io.py", line 1894, in load_program_state
RuntimeError: Failed to load model file , please make sure model file is saved with the following APIs: save_params, save_persistables, save_vars