updates

2020-01-17 11:17:18 -08:00 · 2020-01-17 11:17:18 -08:00 · 831d2b3dcc
parent a8e1390028
commit 831d2b3dcc
1 changed files with 0 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -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: