Restructure repo: moved architecture to model/ package

model/__init__.py

@@ -0,0 +1,4 @@
+from .modeling_bert_ffnn import BERT_FFNN
+from .configuration_bert_ffnn import BertFFNNConfig
+
+__all__ = ["BERT_FFNN", "BertFFNNConfig"]

modeling_bert_ffnn.py → model/modeling_bert_ffnn.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 from transformers import PreTrainedModel, AutoModel
-from .configuration_bert_ffnn import BertFFNNConfig
+from ..model import BertFFNNConfig
 
 
 class AttentionPooling(nn.Module):