spyrit.core.train.tb_profiler

spyrit.core.train.tb_profiler(path_prof, model, criterion, optimizer, dataloader, device, wait=1, warmup=1, active=3, repeat=2)

Tensorboard profiler: Profile code execution