script clean up

neurogenomics · Mar 5, 2024 · a50e75d · a50e75d
1 parent 7dcbde9
commit a50e75d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/README.md b/README.md
@@ -39,7 +39,7 @@ comparisons across them are valid.
 
 ## Reproducing results
 
-The results of our work are based on two models a promoter model (epi_to_express - a custom, convolutional 
+The results of our work are based on two models a promoter model (ChromExpress - a custom, convolutional 
 neural network (CNN)) and a distal model, [chromoformer](https://www.nature.com/articles/s41467-022-34152-5),
 transformer-based, DNA interaction-aware deep learning architecture.
 

diff --git a/train_run_chromoformer.py b/train_run_chromoformer.py
@@ -335,7 +335,4 @@ def get_lr(optimizer):
             if early_stopping.early_stop:
                 break
             #scheduler.step()
-        #save result after 100 epochs or es
-        print('epoch',epoch)
-        print(aa)