Skip to content
This repository has been archived by the owner on Mar 1, 2022. It is now read-only.

Commit

Permalink
Refactoring and version update
Browse files Browse the repository at this point in the history
  • Loading branch information
asiryan committed Nov 8, 2021
1 parent c9acbe7 commit e5be508
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
2 changes: 0 additions & 2 deletions caffe2onnx/src/OPs/Eltwise.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,14 +17,12 @@ def createEltwise(layer, nodename, inname, outname, input_shape):
def __createMul(layer, nodename, inname, outname, input_shape):
output_shape = input_shape[0]
node = Node.c2oNode(layer, nodename, "Mul", inname, outname, input_shape, output_shape)
print(nodename, " node construction completed")
return node

##---------------------Add layer, which may be the addition of two intermediate layer outputs, or it may be an output plus a bias------------------------##
def __createAdd(layer, nodename, inname, outname, input_shape):
output_shape = [input_shape[0]]
node = Node.c2oNode(layer, nodename, "Add", inname, outname, input_shape, output_shape)
print(nodename, " node construction completed")
return node

##----------------------------------------------Max-------------------------------------------------------------##
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

# package metadata
NAME = 'caffe2onnx'
VERSION = '1.1.2'
VERSION = '1.1.3'
DESCRIPTION = 'Convert Caffe models to ONNX.'
LICENSE = 'BSD-3'
GIT = 'https://github.com/asiryan/caffe-onnx'
Expand Down

0 comments on commit e5be508

Please sign in to comment.