Skip to content

Instantly share code, notes, and snippets.

View ArtemisDicoTiar's full-sized avatar
🌟
Moving Forward

John JongYoon Kim ArtemisDicoTiar

🌟
Moving Forward
View GitHub Profile
@kwang2049
kwang2049 / modeling_distilbert.py
Created May 28, 2021 08:13
DistilBERT modeling with LM head supported. One can download it and import modeling_distilbert to support DistilBERT for decoding usage, e.g. TSDAE: https://github.com/UKPLab/sentence-transformers/blob/master/examples/unsupervised_learning/TSDAE/train_tsdae_from_file.py
# coding=utf-8
# Copyright 2019-present, the HuggingFace Inc. team, The Google AI Language Team and Facebook, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software