import torch
import torch.nn as nn
import torch.optim as optim

import numpy as np

class Baseline(nn.Module):
    def __init__(self, hidden_dim, output_dim):
        pass

    def forward(self, input):
        output = None
        return output 
        