#!/usr/bin/env python
# -*- coding: utf-8 -*-

# imports
import torch


class CategoryBranch(torch.nn.Module):

    def __init__(self):
        super(CategoryBranch, self).__init__()
        # TODO: Add the branch layers: CNN(s) followed by RNNs.

    def forward(self, *inputs):
        # TODO: Implement the forward passing
        pass


def main():
    pass


if __name__ == '__main__':
    main()

# EOF
