fix

downloads-generation/models_class1_pan_refined/hyperparameters.yaml

@@ -2,7 +2,7 @@
 # Batch generation
 #########################
 batch_generator_validation_split: 0.1
-batch_generator_batch_size: 10240
+batch_generator_batch_size: 1024
 batch_generator_affinity_fraction: 0.5
 max_epochs: 500
 random_negative_rate: 1.0

mhcflurry/class1_presentation_neural_network.py

@@ -453,7 +453,7 @@ class Class1PresentationNeuralNetwork(object):
 
         allele_representations_hash = self.set_allele_representations(
             allele_representations)
-        loss_reduction = "none"
+        loss_reduction = "sum_over_batch_size"
         self.network.compile(
             loss=[
                 affinities_loss.get_keras_loss(reduction=loss_reduction),

mhcflurry/downloads.yml

@@ -30,7 +30,7 @@ releases:
               default: false
 
             - name: models_class1_pan_refined
-              url: https://github.com/openvax/mhcflurry/releases/download/1.4.0/models_class1_pan_refined.20191211.tar.bz2
+              url: https://github.com/openvax/mhcflurry/releases/download/1.4.0/models_class1_pan_refined.20191211b.tar.bz2
               default: false
 
             - name: models_class1_pan_variants