sgugger
/

my-bert-model

Feature Extraction

Model card Files Files and versions Community

my-bert-model / modeling.py

sgugger's picture

Fix syntax error

1ddc92c over 2 years ago

raw history blame

No virus

179 Bytes

	from transformers import BertModel

	class MyBertModel(BertModel):
	def forward(self, args, *kwargs):
	print("Success")
	return super().forward(args, *kwargs)