deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

some checks

Browse Source
This commit is contained in:
James Betker 2022-06-09 21:46:32 -06:00
parent 34005367fd
commit 07bdd865dc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/trainer/injectors/gaussian_diffusion_injector.py
View File

@ -21,6 +21,7 @@ def masked_channel_balancer(inp, proportion=1):
def channel_restriction(inp, low, high): def channel_restriction(inp, low, high):
assert low > 0 and low < inp.shape[1] and high <= inp.shape[1]
m = torch.zeros_like(inp) m = torch.zeros_like(inp)
m[:,low:high] = 1 m[:,low:high] = 1
return inp * m return inp * m