import torch as t
import torch.nn as nn
import torch.functional as F
import torch.optim as optim


def parse_args():
    # TODO: command-line args for hparams
    pass

def train():
    # TODO: training loop
    pass

def eval():
    pass

def setup():
    # TODO: wandb logging, load configs, all that stuff
    pass

if __name__=="__main__":
    parse_args()