Skip to content

Instantly share code, notes, and snippets.

@isaacmg

isaacmg/column.py

Created May 1, 2020 04:12
Show Gist options
  • Save isaacmg/29709b0a3ddfe4313855736f1165cc24 to your computer and use it in GitHub Desktop.
Save isaacmg/29709b0a3ddfe4313855736f1165cc24 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
column.py
model = model_class.from_pretrained(self.args.model_path,
config=config,
# SpERT model parameters
cls_token=self._tokenizer.convert_tokens_to_ids('[CLS]'),
relation_types=input_reader.relation_type_count - 1,
entity_types=input_reader.entity_type_count,
max_pairs=self.args.max_pairs,
prop_drop=self.args.prop_drop,
size_embedding=self.args.size_embedding,
freeze_transformer=self.args.freeze_transformer)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment