Commit 89a2e5fb authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed an inport bug

parent f63736f4
......@@ -18,7 +18,7 @@ In order to run the network, in the terminal, the user needs to pass it relevant
import torch
from utils.data_utils import get_datasets
import BrainMapperUNet as BrainMapperUNet
from BrainMapperUNet import BrainMapperUNet
import torch.utils.data as data
# Set the default floating point tensor type to FloatTensor
......@@ -57,7 +57,7 @@ def load_data(data_parameters):
return train_data, test_data
def train(data_parameters, training_parameters):
def train(data_parameters, training_parameters, network_parameters):
"""Name
Desc
......@@ -67,14 +67,13 @@ def train(data_parameters, training_parameters):
Train data is also re-shuffled at each training epoch.
Args:
data_parameters(dict):
data_parameters(dict):{
paramters
}
training_parameters(dict):{
paraters
}
network_parameters (dict): Contains information relevant parameters = {
parameters
}
parameters = {
'kernel_heigth': 5
'kernel_width': 5
'kernel_classification': 1
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment