Skip to content
Snippets Groups Projects

Cross validation

Merged Anjith GEORGE requested to merge cross_validation into master
1 unresolved thread
1 file
+ 3
3
Compare changes
  • Side-by-side
  • Inline
@@ -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='.')
Loading