a-ragab-h-m commited on
Commit
f1e66eb
·
verified ·
1 Parent(s): 137d179

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -132,7 +132,7 @@ for epoch in range(params['num_epochs']):
132
  del batch
133
 
134
  # التحقق من الأداء
135
- if epoch % 1 == 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,7 +147,7 @@ for epoch in range(params['num_epochs']):
147
  print('\nNew baseline record\n')
148
 
149
  # التقييم وحفظ النموذج دائمًا
150
- if (epoch % 1 == 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
 
 
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