Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

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