Commit d9bd22ca authored by Christian Chapman-Bird's avatar Christian Chapman-Bird
Browse files

Fixed imports on network train file

parent 71368f54

inference/selection_function/train_network.py

+2 −2
Original line number Diff line number Diff line 
from torch import nn

import torch

import numpy as np

from EMRI_DET.nn.model_train_test import model_train_test

from EMRI_DET.nn.model_creation import create_mlp

from emri_comfi.nn.model_train_test import model_train_test

from emri_comfi.nn.model_creation import create_mlp

import pandas as pd



if __name__ == '__main__':