Skip to content
This repository has been archived by the owner on Nov 22, 2022. It is now read-only.

Commit

Permalink
tie weights in masked lm (#455)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #455

Tie word embedding weights with output layer weights

Reviewed By: kartikayk

Differential Revision: D14686543

fbshipit-source-id: fc15f756b1043b6ab7b208b43442c5c13ef402ea
  • Loading branch information
borguz authored and facebook-github-bot committed Apr 8, 2019
1 parent fb58d20 commit 5f8250c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytext/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def prepare_task(
set_random_seeds(config.random_seed)

if config.load_snapshot_path and os.path.isfile(config.load_snapshot_path):
task = load(config.load_snapshot_path)
task, _ = load(config.load_snapshot_path)
else:
task = create_task(config.task, metadata=metadata)

Expand Down

0 comments on commit 5f8250c

Please sign in to comment.