On_trace_ready ( callable) – callable that is called at each step when schedule ![]() ProfilerAction value that specifies the profiler action to perform at each step. Schedule ( callable) – callable that takes step (int) as a single parameter and returns , .ĭefault value: ProfilerActivity.CPU and (when available) ProfilerActivity.CUDA. profile ( *, activities = None, schedule = None, on_trace_ready = None, record_shapes = False, profile_memory = False, with_stack = False, with_flops = False, with_modules = False, use_cuda = None ) ¶Īctivities ( iterable) – list of activity groups (CPU, CUDA) to use in profiling, supported values: ![]() ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |