Skip to content

Commit

Permalink
Improve Repository Structure (#7)
Browse files Browse the repository at this point in the history
* Add 2 modules:  for building blocks and  for end to end models

* Integrate to existing structure

* Maintain BC
  • Loading branch information
trsvchn authored Oct 1, 2023
1 parent 8d8a692 commit ed7394b
Show file tree
Hide file tree
Showing 5 changed files with 15 additions and 0 deletions.
5 changes: 5 additions & 0 deletions flexynesis/models/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
from .direct_pred import DirectPred
from .supervised_vae import SupervisedVAE
from .triplet_encoder import MultiTripletNetwork

__all__ = ["DirectPred", "SupervisedVAE", "MultiTripletNetwork"]
1 change: 1 addition & 0 deletions flexynesis/models/direct_pred.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from ..model_DirectPred import DirectPred
1 change: 1 addition & 0 deletions flexynesis/models/supervised_vae.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from ..model_SVAE import supervised_vae as SupervisedVAE
1 change: 1 addition & 0 deletions flexynesis/models/triplet_encoder.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from ..model_TripletEncoder import MultiTripletNetwork
7 changes: 7 additions & 0 deletions flexynesis/modules.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
import torch
from torch import nn

from .models_shared import Encoder, Decoder, MLP, EmbeddingNetwork, Classifier
from .model_TripletEncoder import MultiEmbeddingNetwork

__all__ = ["Encoder", "Decoder", "MLP", "EmbeddingNetwork", "MultiEmbeddingNetwork", "Classifier"]

0 comments on commit ed7394b

Please sign in to comment.