Skip to content

Instantly share code, notes, and snippets.

View Nov05's full-sized avatar
💭
Homo Sapiens

Nov05

💭
Homo Sapiens
View GitHub Profile
@Nov05
Nov05 / model.py
Last active March 11, 2020 04:33
2020-03-07 CNN-LSTM image captioning
import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision.models as models
class EncoderCNN(nn.Module):
def __init__(self, embed_size):
# super(EncoderCNN, self).__init__()
super().__init__()