This commit is contained in:
Glenn Jocher 2020-01-17 11:17:18 -08:00
parent a8e1390028
commit 831d2b3dcc
1 changed files with 0 additions and 1 deletions

View File

@ -86,7 +86,6 @@ def train():
# Optimizer
pg0, pg1, pg2 = [], [], [] # optimizer parameter groups
for k, v in dict(model.named_parameters()).items():
print(k)
if '.bias' in k:
pg2 += [v] # biases
elif 'Conv2d.weight' in k: