diff options
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 5 |
1 files changed, 0 insertions, 5 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index 207808ee..2df38c70 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -395,11 +395,6 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log previous_mean_loss = 0
print("Mean loss of {} elements".format(size))
- ititial_step = hypernetwork.step or 0
- if ititial_step > steps:
- return hypernetwork, filename
-
-
weights = hypernetwork.weights()
for weight in weights:
weight.requires_grad = True
|