mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
aa1fd6b45a
Summary: This PR implements UninitializedBuffer and LazyBatchnormXd based on https://github.com/pytorch/pytorch/issues/44538. (cc. emcastillo and albanD) Pull Request resolved: https://github.com/pytorch/pytorch/pull/51548 Reviewed By: zhangguanheng66 Differential Revision: D26276903 Pulled By: albanD fbshipit-source-id: 0ac706974178363f8af075e59b41d5989418922f
6 lines
175 B
Python
6 lines
175 B
Python
from .modules import *
|
|
from .parameter import Parameter, UninitializedParameter, UninitializedBuffer
|
|
from .parallel import DataParallel
|
|
from . import init
|
|
from . import utils
|