Closed JaehyeokLee-119 closed 1 year ago
trainer.py 380줄, 248줄 2번째 utterance에 대한 pair_window_idx를 구할 때 1번째 utterance의 영역을 건드리는 문제
개선 예시) pair_window_idx[max(0, int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1 -> pair_window_idx[max(0, int((ent_t-1)*end_t)/2), int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1
pair_window_idx[max(0, int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1
pair_window_idx[max(0, int((ent_t-1)*end_t)/2), int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1
수정 완료
trainer.py 380줄, 248줄 2번째 utterance에 대한 pair_window_idx를 구할 때 1번째 utterance의 영역을 건드리는 문제
개선 예시)
pair_window_idx[max(0, int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1
->pair_window_idx[max(0, int((ent_t-1)*end_t)/2), int(end_t * (end_t + 1) / 2) - window_constraint):int(end_t * (end_t + 1) / 2)] = 1