[发布] 打包发布

lightGE/core/__init__.py

@@ -1,2 +1,7 @@
-from lightGE.core.nn import *
-from lightGE.core.tensor import *
+from lightGE.core.nn import Model, Sequential, Linear, Conv2d, MaxPool2d, AvgPool2d, LSTM, Tanh, Sigmoid, ReLu, \
+    BatchNorm1d, BatchNorm2d, Dropout, Dropout2d
+from lightGE.core.tensor import Tensor, Op, AddOp, SubOp, MulOp, DivOp, PowOp, NegOp, MatMulOp, SumOp, MeanOp, \
+    TransposeOp, ReshapeOp, SqueezeOp, UnsqueezeOp, LogOp, ExpOp, \
+    SoftmaxOp, AbsOp, MaxOp, VarOp, SqrtOp, ReLuOp, Conv2dOp, CosOp, SinOp, MaxPool2dOp, AvgPool2dOp, \
+    log, exp, abs, max, var, sqrt, cos, sin, sum, mean, softmax, tanh, sigmoid, relu, conv2d, max_pool2d, avg_pool2d, \
+    TcGraph

lightGE/data/__init__.py

@@ -1 +1 @@
-from lightGE.data.dataloader import *
+from lightGE.data.dataloader import Dataset, DataLoader

lightGE/utils/__init__.py

@@ -1,4 +1,4 @@
-from lightGE.utils.loss import *
-from lightGE.utils.optimizer import *
-from lightGE.utils.scheduler import *
-from lightGE.utils.trainer import *
+from lightGE.utils.loss import mseLoss, maeLoss, crossEntropyLoss, huberLoss, nll_loss
+from lightGE.utils.optimizer import Optimizer, SGD, Adam, AdaGrad, RMSprop, SGDMomentum
+from lightGE.utils.scheduler import Scheduler, MultiStepLR, StepLR, Exponential, Cosine, LambdaLR, ReduceLROnPlateau
+from lightGE.utils.trainer import Trainer

setup.py

@@ -0,0 +1,8 @@
+from distutils.core import setup
+
+setup(
+    name='LightGE',
+    version='0.1',
+    packages=['lightGE', 'LightGE.core', 'LightGE.data', 'LightGE.utils'],
+    url='https://gitlab.com/l7829936101/nnengine'
+)