From 7da4f0485032bb8b8acfc678546ffcea3a23a44b Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 21 Apr 2023 11:43:50 +0200
Subject: Update

---
 training/functional.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'training/functional.py')

diff --git a/training/functional.py b/training/functional.py
index c6ceb20..695a24f 100644
--- a/training/functional.py
+++ b/training/functional.py
@@ -456,7 +456,7 @@ def train_loop(
     sample_frequency: int = 10,
     checkpoint_frequency: int = 50,
     milestone_checkpoints: bool = True,
-    cycle: int = 1,
+    cycle: int = 0,
     global_step_offset: int = 0,
     num_epochs: int = 100,
     gradient_accumulation_steps: int = 1,
@@ -537,7 +537,7 @@ def train_loop(
 
             logs = {}
 
-            with on_train(epoch):
+            with on_train(cycle):
                 for step, batch in enumerate(train_dataloader):
                     loss, acc, bsz = loss_step(step, batch, cache)
                     loss /= gradient_accumulation_steps
-- 
cgit v1.2.3-54-g00ecf