Spaces:

MukeshKapoor25
/

logbert_processor

Runtime error

logbert_processor / bert_pytorch /model /embedding /segment.py

changs

6f2ff70 over 1 year ago

157 Bytes

import torch.nn as nn


class SegmentEmbedding(nn.Embedding):
    def __init__(self, embed_size=512):
        super().__init__(3, embed_size, padding_idx=0)

	import torch.nn as nn


	class SegmentEmbedding(nn.Embedding):
	def __init__(self, embed_size=512):
	super().__init__(3, embed_size, padding_idx=0)