| 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() |