Skip to content

Commit

Permalink
consistent with hybrid depth
Browse files Browse the repository at this point in the history
keep hyper-parameter meaning consistent for convenience to compare in the training record
  • Loading branch information
TITC authored May 21, 2022
1 parent d1f21f9 commit 0938894
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pix2tex/models/vit.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ def get_encoder(args):
emb_dropout=args.get('emb_dropout', 0),
attn_layers=Encoder(
dim=args.dim,
depth=args.num_layers,
depth=args.encoder_depth,
heads=args.heads,
)
)

0 comments on commit 0938894

Please sign in to comment.