harari/evo2-finetuned-chromosome-classifier-epoch20

EVO2 Finetuned Chromosome Classifier (Epoch 20)

This is a finetuned EVO2 model for chromosome classification, trained for 20 epochs.

Model Details

• Base Model: EVO2
• Task: Chromosome Classification
• Training: 20 epochs
• Model Size: 65MB
• Framework: PyTorch

Usage

import torch
import torch.nn as nn

# Load the model
model = torch.load('model_epoch_20.pth', map_location='cpu')
model.eval()

# For inference
with torch.no_grad():
    # Your input tensor here
    output = model(input_tensor)
    predictions = torch.sigmoid(output)

Training Details

The model was finetuned on chromosome data across all chromosomes (experiment 9).

Files

• model_epoch_20.pth: The PyTorch model checkpoint
• inference.py: Sample inference code
• extract_predictions.py: Script to extract predictions

Citation

If you use this model, please cite:

• Original EVO2 paper
• This finetuned model repository

License

Apache 2.0