fix for #6700
This commit is contained in:
parent
e452facef4
commit
11183b4d90
|
@ -115,7 +115,7 @@ class PersonalizedBase(Dataset):
|
|||
weight /= weight.mean()
|
||||
elif use_weight:
|
||||
#If an image does not have a alpha channel, add a ones weight map anyway so we can stack it later
|
||||
weight = torch.ones([channels] + latent_size)
|
||||
weight = torch.ones(latent_sample.shape)
|
||||
else:
|
||||
weight = None
|
||||
|
||||
|
|
Loading…
Reference in New Issue