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 35db4c1 commit f10cd86Copy full SHA for f10cd86
deepxde/utils/internal.py
@@ -19,12 +19,14 @@ def timing(f):
19
@wraps(f)
20
def wrapper(*args, **kwargs):
21
verbose = kwargs.get('verbose', 1)
22
- ts = timeit.default_timer()
23
- result = f(*args, **kwargs)
24
- te = timeit.default_timer()
25
if verbose > 0 and config.rank == 0:
+ ts = timeit.default_timer()
+ result = f(*args, **kwargs)
+ te = timeit.default_timer()
26
print("%r took %f s\n" % (f.__name__, te - ts))
27
sys.stdout.flush()
28
+ else:
29
30
return result
31
32
return wrapper
0 commit comments