add extra block, lower feature sizes

2023-01-25 23:10:51 +00:00
parent 46e37e22eb
commit 9cd9c6f12a
6 changed files with 30 additions and 19 deletions
@@ -177,7 +177,10 @@ def train():
        if step == 0:
            to_log = { **to_log, **{ 'model/summary': wandb.Html(get_summary_html(model)) } }
 
-        wandb.log(to_log, step=step)  
+        if step % 1000 == 0:    
+            wandb.log(to_log, step=step, commit=True)
+        else:
+            wandb.log(to_log, step=step, commit=False)  
        # print(step)

        if step % 10000 == 0: