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):


 "reinforce.py", line 164, in <module>

main()


 "reinforce.py", line 158, in main

reinforce.run()


 "reinforce.py", line 57, in run

self.dialog.run(ctxs, self.logger)


 "/home/ss8464/dl/nego_comm/main/src/dialog.py", line 170, in run

out = writer.write()


 "/home/ss8464/dl/nego_comm/main/src/agent.py", line 128, in write

100, self.args.temperature)


 "/home/ss8464/dl/nego_comm/main/src/models/dialog_model.py", line 300, in write

prob = F.softmax(scores)


 "/share/apps/pytorch/20171124/python3.6/gnu/lib/python3.6/site-packages/torch/nn/functional.py", line 818, in softmax

return torch._C._nn.softmax(input, dim)
RuntimeError: dimension out of range (expected to be in range of [-1, 0], but got 1)`