Newer
Older
from src.options import TrainingOptions
from src.data import CustomDatasetDataLoader
from src.models import create_model, train_model
if __name__ == "__main__":
print("-----------------Networks---------------")
print(model)
print("---------------End networks-------------")