TRL 文档
其他
加入 Hugging Face 社区
并获得增强的文档体验
开始使用
其它
profiling_decorator
trl.extras.profiling.profiling_decorator
< 源代码 >( func: <built-in function 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 @ B
profiling_context
trl.extras.profiling.profiling_context
< 源代码 >( trainer: Trainer name: str )
一个用于分析代码块性能的上下文管理器函数。结果将根据 trainer 的配置记录到 Weights & Biases 或 MLflow 中。
示例
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