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

modified unittest

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