TRL 文档
其他
加入 Hugging Face 社区
并获得增强的文档体验
开始入门
其他
profiling_decorator
trl.extras.profiling.profiling_decorator
< 源代码 >( func: <内置函数 callable> )
用于分析函数性能并使用 extras.profiling.profiling_context() 记录执行时间的装饰器。
示例
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
< 源代码 >( trainer: Trainer name: str )
用于分析代码块性能的上下文管理器函数。如果启用，结果将记录到 Weights & Biases。
示例
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