Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
4446c02b
Commit
4446c02b
authored
Oct 07, 2019
by
Rémi Louf
Browse files
add wireframe for seq2seq model
parent
904158ac
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
60 additions
and
0 deletions
+60
-0
transformers/modeling_seq2seq.py
transformers/modeling_seq2seq.py
+37
-0
transformers/tests/modeling_seq2seq_test.py
transformers/tests/modeling_seq2seq_test.py
+23
-0
No files found.
transformers/modeling_seq2seq.py
0 → 100644
View file @
4446c02b
# coding=utf-8
# Copyright 2018 The HuggingFace Inc. team.
#
# 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
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" Conditional generation class. """
class
Seq2SeqModel
(
object
):
def
__init__
(
self
):
# need to use from_pretrained to initialize
raise
NotImplementedError
@
classmethod
def
from_pretrained
(
cls
,
encoder
,
decoder
):
# Here we should call AutoModel to initialize the models depending
# on the pretrained models taken as an input.
# For a first iteration we only work with Bert.
raise
NotImplementedError
def
__call__
(
self
):
# allows to call an instance of the class
# model = Seq2Seq(encode='bert', decoder='bert')
raise
NotImplementedError
def
process
(
self
):
# alternative API to __call__ it is more explicit.
raise
NotImplementedError
transformers/tests/modeling_seq2seq_test.py
0 → 100644
View file @
4446c02b
# coding=utf-8
# Copyright 2018 The Google AI Language Team Authors.
#
# 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
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import
unittest
class
Seq2SeqTest
(
unittest
.
TestCase
):
raise
NotImplementedError
def
__main__
():
unittest
.
main
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment