Spaces:
Running
Running
add frcrn model
Browse files
examples/frcrn/step_2_train_model.py
CHANGED
@@ -175,7 +175,7 @@ def main():
|
|
175 |
step_idx = int(step_idx)
|
176 |
if step_idx > last_step_idx:
|
177 |
last_step_idx = step_idx
|
178 |
-
last_epoch = 1
|
179 |
|
180 |
if last_step_idx != -1:
|
181 |
logger.info(f"resume from steps-{last_step_idx}.")
|
|
|
175 |
step_idx = int(step_idx)
|
176 |
if step_idx > last_step_idx:
|
177 |
last_step_idx = step_idx
|
178 |
+
# last_epoch = 1
|
179 |
|
180 |
if last_step_idx != -1:
|
181 |
logger.info(f"resume from steps-{last_step_idx}.")
|