CoolFace
Modelpublic

CluelessNovice/demo_cls_1

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes5downloads
Model Card

basemodel: westlake-repl/SaProt35M_AF2

library_name: peft


Model Card for Model-demo-35M

This model is used for a demo classification task

Task type

Protein-level Classification

Model input type

SA Sequence

Label meanings

0: A

1: B

LoRA config

  • —r: 8
  • —lora_dropout: 0.0
  • —lora_alpha: 16
  • —target_modules: ['key', 'query', 'output.dense', 'intermediate.dense', 'value']
  • —modules_to_save: ['classifier']

Training config

  • —optimizer:
  • —class: AdamW
  • —betas: (0.9, 0.98)
  • —weight_decay: 0.01
  • —learning rate: 0.001
  • —epoch: 1
  • —batch size: 2
  • —precision: 16-mixed