Skip to content
Snippets Groups Projects

Cross validation

Merged Anjith GEORGE requested to merge cross_validation into master
1 unresolved thread
1 file
+ 2
1
Compare changes
  • Side-by-side
  • Inline
@@ -225,7 +225,8 @@ def test_MCCNNtrainer():
@@ -225,7 +225,8 @@ def test_MCCNNtrainer():
from ..architectures import MCCNN
from ..architectures import MCCNN
net = MCCNN(num_channels=4)
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
from ..trainers import MCCNNTrainer
trainer = MCCNNTrainer(net, verbosity_level=3)
trainer = MCCNNTrainer(net, verbosity_level=3)
Loading