提交 8c70c4e4 编写于 作者: Z zhengya01

update debug info for dialogue_general_understanding

上级 fc0d9a18
...@@ -361,7 +361,6 @@ def main(args): ...@@ -361,7 +361,6 @@ def main(args):
break break
if args.do_train and args.enable_ce: if args.do_train and args.enable_ce:
card_num = get_cards() card_num = get_cards()
print("zytest_card_num", card_num)
ce_loss = 0 ce_loss = 0
ce_acc = 0 ce_acc = 0
ce_time = 0 ce_time = 0
...@@ -402,7 +401,6 @@ def main(args): ...@@ -402,7 +401,6 @@ def main(args):
def get_cards(): def get_cards():
num = 0 num = 0
cards = os.environ.get('CUDA_VISIBLE_DEVICES', '') cards = os.environ.get('CUDA_VISIBLE_DEVICES', '')
print("zytest_cards", cards)
if cards != '': if cards != '':
num = len(cards.split(",")) num = len(cards.split(","))
return num return num
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册