TRL documentation
Other
Other
profiling_decorator
trl.extras.profiling.profiling_decorator
< source >( func: Callable ) → Callable
Decorator to profile a function and log execution time using extras.profiling.profiling_context().
This decorator works with methods that have access to a trainer instance (typically as self).
Example:
from transformers import Trainer
from trl.extras.profiling import profiling_decorator
class MyTrainer(Trainer):
@profiling_decorator
def some_method(self):
A = np.random.rand(1000, 1000)
B = np.random.rand(1000, 1000)
# Code to profile: simulate a computationally expensive operation
result = A @ Bprofiling_context
trl.extras.profiling.profiling_context
< source >( trainer: Trainer name: str ) → ProfilingContext
Factory function to create a ProfilingContext from a Trainer instance.
This function maintains backwards compatibility with existing code while using the decoupled ProfilingContext class internally.
Example:
from transformers import Trainer
from trl.extras.profiling import profiling_context
class MyTrainer(Trainer):
def some_method(self):
A = np.random.rand(1000, 1000)
B = np.random.rand(1000, 1000)
with profiling_context(self, "matrix_multiplication"):
# Code to profile: simulate a computationally expensive operation
result = A @ B # Matrix multiplication