Skip to content

Instantly share code, notes, and snippets.

View paduvi's full-sized avatar

Phan Đức Việt paduvi

View GitHub Profile
@paduvi
paduvi / FlatCnnLayer.py
Last active May 13, 2023 20:30
Hierarchical Softmax CNN Classification
import torch
import torch.nn as nn
import torch.nn.init as init
dropout_prob = 0.5
class FlatCnnLayer(nn.Module):
def __init__(self, embedding_size, sequence_length, filter_sizes=[3, 4, 5], out_channels=128):
super(FlatCnnLayer, self).__init__()

TodoMVC React