Update run.py
Browse files
run.py
CHANGED
@@ -132,7 +132,7 @@ for epoch in range(params['num_epochs']):
|
|
132 |
del batch
|
133 |
|
134 |
# التحقق من الأداء
|
135 |
-
if epoch %
|
136 |
baseline_dataloader = DataLoader(baseline_dataset, batch_size=batch_size, collate_fn=baseline_dataset.collate)
|
137 |
tot_cost = []
|
138 |
for batch in baseline_dataloader:
|
@@ -147,7 +147,7 @@ for epoch in range(params['num_epochs']):
|
|
147 |
print('\nNew baseline record\n')
|
148 |
|
149 |
# التقييم وحفظ النموذج دائمًا
|
150 |
-
if (epoch %
|
151 |
b = max(int(batch_size // sample_size**2), 1)
|
152 |
validation_dataloader = DataLoader(validation_dataset, batch_size=b, collate_fn=validation_dataset.collate)
|
153 |
|
|
|
132 |
del batch
|
133 |
|
134 |
# التحقق من الأداء
|
135 |
+
if epoch % 5 == 0:
|
136 |
baseline_dataloader = DataLoader(baseline_dataset, batch_size=batch_size, collate_fn=baseline_dataset.collate)
|
137 |
tot_cost = []
|
138 |
for batch in baseline_dataloader:
|
|
|
147 |
print('\nNew baseline record\n')
|
148 |
|
149 |
# التقييم وحفظ النموذج دائمًا
|
150 |
+
if (epoch % 10 == 0) and run_tests:
|
151 |
b = max(int(batch_size // sample_size**2), 1)
|
152 |
validation_dataloader = DataLoader(validation_dataset, batch_size=b, collate_fn=validation_dataset.collate)
|
153 |
|