Update learned_round_optimizer.py

Xilinx · Dec 5, 2024 · 0f0dd43 · 0f0dd43
1 parent d369baa
commit 0f0dd43
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/brevitas_examples/common/learned_round/learned_round_optimizer.py b/src/brevitas_examples/common/learned_round/learned_round_optimizer.py
@@ -693,7 +693,6 @@ def skip_full_execution(self, block, next_block, block_forward, cache):
         # The second is a quantized one to create the quantized input of the next blocks
 
         # We use the cache output to generate a new temporary dataloder for the next block
-        # and to update our floating_point_dataset
         tmp_data_loader = []
         for i in range(len(cache)):
             (args, kwargs), output = cache.sample_batch([i])