Bert_GradCAM

• Bert를 이용한 준지도학습 모델에 GradCAM을 추가한 실험입니다.

How to use

python main.py --max_len 512 --epochs 1 --batch 16 --label_batch 200 --masked True --xai True --test_num 10000 --class_num 100 --gpu_set choose[1,2,3,4]

Model Framework

• 아래 Text Self training with XAI 실험에서 Bert+CNN 과 GradCAM을 활용한 코드 입니다.