We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7300783 commit 20f4189Copy full SHA for 20f4189
conf.py
@@ -112,17 +112,13 @@
112
def reset_seeds(gallery_conf, fname):
113
global torch
114
torch._dynamo.reset()
115
- torch.cuda.empty_cache()
116
- if torch.cuda.is_available():
117
118
- torch.cuda.reset_peak_memory_stats()
+ torch.dynamo.reset()
+ torch.cuda.empty_cache()
119
torch.manual_seed(42)
120
torch.set_default_device(None)
121
random.seed(10)
122
numpy.random.seed(10)
123
gc.collect()
124
- import torch._inductor.utils
125
- torch._inductor.utils.clear_inductor_caches()
126
127
128
sphinx_gallery_conf = {
0 commit comments