Spaces:

Muthukamalan
/

UnsolvedMNIST

Sleeping

Muthukamalan commited on Jul 10, 2024

Commit

8de6687

1 Parent(s): 91820d0

train script added

Files changed (1) hide show

train.py CHANGED Viewed

@@ -122,7 +122,7 @@ summary(
 trainer  = pl.Trainer(
                 max_epochs=CONFIG['training'].get('num_epochs',15),
                 logger=logger,
-                profiler=perf_profiler,
                 callbacks=call_backs,
                 precision=32,
                 enable_model_summary=False,

 trainer  = pl.Trainer(
                 max_epochs=CONFIG['training'].get('num_epochs',15),
                 logger=logger,
+                profiler=perf_profiler,#'advanced',
                 callbacks=call_backs,
                 precision=32,
                 enable_model_summary=False,