python interface
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
|
||||
import sys
|
||||
sys.path.append('../')
|
||||
from core.tikzeng import *
|
||||
|
||||
# defined your arch
|
||||
arch = [
|
||||
to_head( '..' ),
|
||||
to_cor(),
|
||||
to_begin(),
|
||||
to_Conv( 'conv1', 572, 64, offset="(0,0,0)", to="(0,0,0)" ),
|
||||
to_Pool("pool1", offset="(0,0,0)", to="(conv1-east)"),
|
||||
to_SoftMax( "soft1", "SOFT", 10 ,"(3,0,0)", "(pool1-east)" ),
|
||||
to_connection( "pool1", "soft1"),
|
||||
to_end()
|
||||
]
|
||||
|
||||
def main():
|
||||
namefile = str(sys.argv[0]).split('.')[0]
|
||||
to_generate(arch, namefile + '.tex' )
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user