Some faster/better kernels. Tensors with points are now kept in (b, c, ...) format as this is easier for pytorch

2026-08-20 12:00:32 +08:00 · 2018-02-10 20:32:52 -05:00
parent 8bce353da4
commit 65a127f3d2
19 changed files with 372 additions and 296 deletions
@@ -144,8 +144,8 @@ if __name__ == "__main__":
        model,
        model_fn,
        optimizer,
-        checkpoint_name="checkpoints/cls_xyz",
-        best_name="checkpoints/cls_xyz_best",
+        checkpoint_name="checkpoints/single_layer",
+        best_name="checkpoints/single_layer_best",
        lr_scheduler=lr_scheduler,
        bnm_scheduler=bnm_scheduler
    )