From 41b1b170530808caf7d68b2fa85008e83c230c4d Mon Sep 17 00:00:00 2001
From: Zhenghao <zfei@zju.edu.cn>
Date: Thu, 18 Jan 2024 10:55:09 +0800
Subject: [PATCH] fix build_encoder bug

---
 detr/models/detr_vae.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detr/models/detr_vae.py b/detr/models/detr_vae.py
index 8c193529..4ebbf68f 100644
--- a/detr/models/detr_vae.py
+++ b/detr/models/detr_vae.py
@@ -282,7 +282,7 @@ def build(args):
     if args.no_encoder:
         encoder = None
     else:
-        encoder = build_transformer(args)
+        encoder = build_encoder(args)
 
     model = DETRVAE(
         backbones,