Update file trainer.cli.py
Browse files- trainer.cli.py +0 -14
trainer.cli.py
CHANGED
|
@@ -18,8 +18,6 @@ parser = ArgumentParser(
|
|
| 18 |
)
|
| 19 |
|
| 20 |
|
| 21 |
-
import torch
|
| 22 |
-
import torch.utils.bottleneck
|
| 23 |
|
| 24 |
|
| 25 |
if __name__ == '__main__':
|
|
@@ -53,19 +51,7 @@ if __name__ == '__main__':
|
|
| 53 |
config.trainer.model = model
|
| 54 |
config.trainer.wandb = wandb
|
| 55 |
|
| 56 |
-
# Create a bottleneck profiler
|
| 57 |
-
profiler = torch.utils.bottleneck.Profiler()
|
| 58 |
-
|
| 59 |
-
# Start profiling
|
| 60 |
-
profiler.start()
|
| 61 |
|
| 62 |
|
| 63 |
trainer = Trainer(config.trainer)
|
| 64 |
trainer.train(batches)
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
# Stop profiling
|
| 68 |
-
profiler.stop()
|
| 69 |
-
|
| 70 |
-
# Retrieve profiling data
|
| 71 |
-
profiler.report()
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
|
|
|
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
if __name__ == '__main__':
|
|
|
|
| 51 |
config.trainer.model = model
|
| 52 |
config.trainer.wandb = wandb
|
| 53 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
|
| 55 |
|
| 56 |
trainer = Trainer(config.trainer)
|
| 57 |
trainer.train(batches)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|