Skip to content

[replace usage of tensor.storage()] #95

Description

@Oran-Ac

Many thanks for your fantastic project. Just a kind suggestion: while using bmtrain, it will throw out a lot warnings and they are all corresponding to torch.storage as following:

/opt/conda/envs/compression/lib/python3.10/site-packages/bmtrain/store.py:178: UserWarning: TypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage()

it seems that it is related to feature of torch. You may replace all the tensor.storage() with tensor.untyped_storage() so that we can get the clean logs:)

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions