Skip to content
Snippets Groups Projects
Select Git revision
  • 7600038f1c865b78204f7c48a308d39facec64d2
  • master default protected
  • csv_export
  • ndex
  • v1.1.18-rc2
  • v1.1.17
  • v1.1.16
  • v1.1.16-rc12
  • v1.1.16-rc11
  • v1.1.16-rc10
  • v1.1.16-rc9
  • v1.1.16-rc8
  • v1.1.16-rc7
  • v1.1.16-rc4
  • v1.1.16-rc3
  • v1.1.16-rc1
  • v1.1.6-rc1
  • v1.1.15
  • v1.1.15-rc7
  • v1.1.15-rc6
  • v1.1.15-rc3
  • v1.1.15-rc1
  • v1.1.14
  • v1.1.13
24 results

analysis-window.component.html

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    test_classifier.py 433 B
    from pytorch_lightning import Trainer, seed_everything
    
    from project.lit_classifier_main import LitClassifier
    
    
    def test_lit_classifier():
        seed_everything(1234)
    
        model = LitClassifier()
        train, val, test = mnist()
        trainer = Trainer(limit_train_batches=50, limit_val_batches=20, max_epochs=2)
        trainer.fit(model, train, val)
    
        results = trainer.test(test_dataloaders=test)
        assert results[0]['test_acc'] > 0.7