vision mlp model based on oneflow
Project description
vision-mlp-oneflow
Vision MLP Models Based on OneFlow
TODO LIST
复现过程中遇到的问题
tensor.chunk()
方法没有和torch.tensor.chunk()
对齐, 在输入的dim参数为负数的时候未作判断, 并且整体功能还变成了repeat操作- 缺少
tensor.floor_()
方法, 只能用tensor = tensor.floor()
替代 - 缺少
tensor.erfinv()
方法及其inplace版本, 无法复现trunc_norm_
- 缺少
tensor.new_empty()
方法 flow.shape
返回的是oneflow._oneflow_internal.Size
无法和tuple
相加, 按以下方法使用会报错:
# oneflow里报错
x = flow.tensor(np.random.randn(1, 16, 512), dtype=flow.float32)
x.shape + (4, )
x.size() + (4, )
>>>TypeError: unsupported operand type(s) for +: 'oneflow._oneflow_internal.Size' and 'tuple'
# torch里不报错
x = torch.randn(1, 16, 512)
x.shape + (4, )
x.size() + (4, )
>>> torch.Size([1, 16, 512, 4])
tensor.gather()
未具体判定输入的dim为负数的情况
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for vision_mlp_oneflow-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1a482cc628e6e486f5e89f0053d61df8a0ee5bf2444eea57079a708c637d448b |
|
MD5 | d87081d3060bb73c8102fc0fa5b5792a |
|
BLAKE2b-256 | f4f619d612ce126f24c1f5e06e19db45ea079d2497a1768035d0b7b9d70a7d31 |