mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
c3a4468af6
Summary: Add conv helpers, the migration of functions assumes that people should not do cnn_model = CNNModelHelper(use_cudnn=True) cnn_model.Conv(..., use_cudnn=False, ...) Reviewed By: salexspb Differential Revision: D4884974 fbshipit-source-id: 12af6e2a5863eba789232cd4a4771f95d05f9227
21 lines
749 B
Python
21 lines
749 B
Python
## @package model_helpers
|
|
# Module caffe2.python.model_helpers
|
|
## @package model_helper_api
|
|
# Module caffe2.python.model_helper_api
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
# from caffe2.python.helpers.arg_scope import *
|
|
# flake8: noqa
|
|
from caffe2.python.helpers.dropout import *
|
|
from caffe2.python.helpers.fc import *
|
|
from caffe2.python.helpers.pooling import *
|
|
from caffe2.python.helpers.normalization import *
|
|
from caffe2.python.helpers.nonlinearity import *
|
|
from caffe2.python.helpers.array_helpers import *
|
|
from caffe2.python.helpers.algebra import *
|
|
from caffe2.python.helpers.train import *
|
|
from caffe2.python.helpers.conv import *
|