model.generate()

개발하는 G0·2024년 4월 2일
PyTorch
0
post-custom-banner

https://github.com/huggingface/transformers/issues/25063

decoder 모델이나 seq2seq 모델처럼 생성을 하는 모델에서 사용한다.
encoder 모델에서는 작동하지 않는다.

profile
개발하는 G0
초보 개발자
이전 포스트

model.zero_grad() vs optimizer.zero_grad()

다음 포스트

model.eval() vs with torch.no_grad():

0개의 댓글