poetry

Author: ZouJiu1

PatchEmbed.py

@@ -122,11 +122,11 @@ def restore_model(self, models):
         self.convolution.restore_model(models[-1])
 
 class Position_Embedding(Embedding_layer):
-    def __init__(self, context_length, vocab_size,  embed_dim, adam = False):
+    def __init__(self, context_length, vocab_size,  embed_dim, adam = False, float32=False, float16 = False):
         self.context_length = context_length
-        self.text_embedding = Embedding_layer(vocab_size, embedding_dim = embed_dim, adam = adam)
+        self.text_embedding = Embedding_layer(vocab_size, embedding_dim = embed_dim, adam = adam, float32=float32, float16=float16)
         # self.pos_embedding  = Position_Fixed(context_length, embed_dim)
-        self.pos_embedding = Embedding_layer(context_length, embedding_dim = embed_dim, adam = adam)
+        self.pos_embedding = Embedding_layer(context_length, embedding_dim = embed_dim, adam = adam, float32=float32, float16=float16)
         self.adam = adam
 
     def forward(self, inputs):

README.md

@@ -0,0 +1,134 @@
+{
+  "id2char":{
+    "0":" ",
+    "1":"!",
+    "2":"\"",
+    "3":"#",
+    "4":"$",
+    "5":"%",
+    "6":"&",
+    "7":"'",
+    "8":"(",
+    "9":")",
+    "10":"*",
+    "11":"+",
+    "12":",",
+    "13":"-",
+    "14":".",
+    "15":"/",
+    "16":"0",
+    "17":"1",
+    "18":"2",
+    "19":"3",
+    "20":"4",
+    "21":"5",
+    "22":"6",
+    "23":"7",
+    "24":"8",
+    "25":"9",
+    "26":":",
+    "27":";",
+    "28":"<",
+    "29":"=",
+    "30":">",
+    "31":"?",
+    "32":"@",
+    "33":"[",
+    "34":"]",
+    "35":"_",
+    "36":"`",
+    "37":"a",
+    "38":"b",
+    "39":"c",
+    "40":"d",
+    "41":"e",
+    "42":"f",
+    "43":"g",
+    "44":"h",
+    "45":"i",
+    "46":"j",
+    "47":"k",
+    "48":"l",
+    "49":"m",
+    "50":"n",
+    "51":"o",
+    "52":"p",
+    "53":"q",
+    "54":"r",
+    "55":"s",
+    "56":"t",
+    "57":"u",
+    "58":"v",
+    "59":"w",
+    "60":"x",
+    "61":"y",
+    "62":"z",
+    "63":"~"
+  },
+  "char2id":{
+    " ":0,
+    "!":1,
+    "\"":2,
+    "#":3,
+    "$":4,
+    "%":5,
+    "&":6,
+    "'":7,
+    "(":8,
+    ")":9,
+    "*":10,
+    "+":11,
+    ",":12,
+    "-":13,
+    ".":14,
+    "/":15,
+    "0":16,
+    "1":17,
+    "2":18,
+    "3":19,
+    "4":20,
+    "5":21,
+    "6":22,
+    "7":23,
+    "8":24,
+    "9":25,
+    ":":26,
+    ";":27,
+    "<":28,
+    "=":29,
+    ">":30,
+    "?":31,
+    "@":32,
+    "[":33,
+    "]":34,
+    "_":35,
+    "`":36,
+    "a":37,
+    "b":38,
+    "c":39,
+    "d":40,
+    "e":41,
+    "f":42,
+    "g":43,
+    "h":44,
+    "i":45,
+    "j":46,
+    "k":47,
+    "l":48,
+    "m":49,
+    "n":50,
+    "o":51,
+    "p":52,
+    "q":53,
+    "r":54,
+    "s":55,
+    "t":56,
+    "u":57,
+    "v":58,
+    "w":59,
+    "x":60,
+    "y":61,
+    "z":62,
+    "~":63
+  }
+}