|
|
@@ -455,7 +455,7 @@ class CollaborativeOptimizer(DecentralizedOptimizerBase):
|
|
|
|
|
|
for state in valid_peer_states:
|
|
|
total_samples_per_second += state.samples_per_second
|
|
|
- if state.step >= global_optimizer_step:
|
|
|
+ if state.step == global_optimizer_step:
|
|
|
total_samples_accumulated += state.samples_accumulated
|
|
|
estimated_current_samples += (
|
|
|
state.samples_accumulated + max(0, current_time - state.time) * state.samples_per_second
|
justheuristic