oneflow
oneflow copied to clipboard
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
绕过 dataloader delete_shm 报错 StopIteration 的问题
eager amp 支持 scaler ,eager 可以做完整的 amp 训练。 https://github.com/Oneflow-Inc/OneTeam/issues/1754 export ONEFLOW_VM_COMPUTE_ON_WORKER_THREAD=0之后 eager amp测试结果:   无论是训练速度还是显存占用相比于fp32模式都有较大的提升。
Add fused Geglu operator, current only implement the forward logic
## Summary import oneflow as torch "RuntimeError: This package is a placeholder. Please install oneflow following the instructions" ## Code to reproduce bug import oneflow as torch ## System Information...
### 本 pr 去掉 Block 上的 attribute 和 config - 1、彻底避免重名问题; - 2、去掉 block config; ``` import flow.nn.Graph as G from oneflow.nn.graph import ModuleGraph from oneflow.nn.module import Module class AGraph(G):...
带作用域的mock
Feature: merge multi tensors into a contiguous tensor buffer according to their dtypes and devices to accelerate model grad updating. Performance tests: https://github.com/Oneflow-Inc/OneTeam/issues/1698#issuecomment-1312489487