Skip to content
Snippets Groups Projects
Commit b5c74e26 authored by Anjith GEORGE's avatar Anjith GEORGE Committed by Anjith GEORGE
Browse files

modified unittest

parent 4a5b2a1f
Branches
Tags
1 merge request!22Cross validation
......@@ -225,7 +225,8 @@ def test_MCCNNtrainer():
from ..architectures import MCCNN
net = MCCNN(num_channels=4)
dataloader = torch.utils.data.DataLoader(DummyDataSetMCCNN(), batch_size=32, shuffle=True)
dataloader={}
dataloader{'train'} = torch.utils.data.DataLoader(DummyDataSetMCCNN(), batch_size=32, shuffle=True)
from ..trainers import MCCNNTrainer
trainer = MCCNNTrainer(net, verbosity_level=3)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment