RuolinZheng08 / twewy-discord-chatbot

Discord AI Chatbot using DialoGPT, trained on the game transcript of The World Ends With You
https://www.freecodecamp.org/news/discord-ai-chatbot/
MIT License
317 stars 156 forks source link

Why inputs and labels are same while training? #12

Open vaishalishrivastava opened 2 years ago

vaishalishrivastava commented 2 years ago
for step, batch in enumerate(epoch_iterator):

            # Skip past any already trained steps if resuming training
            if steps_trained_in_current_epoch > 0:
                steps_trained_in_current_epoch -= 1
                continue

            inputs, labels = (batch, batch)
            if inputs.shape[1] > 1024: continue
            inputs = inputs.to(args.device)
            labels = labels.to(args.device)
            model.train()
            outputs = model(inputs, labels=labels)
whiteeat commented 2 years ago

Yeah, I spot that too. I also observe other repos of training chat bot based on gpt2 and they do the similar thing. I really doubt AI can learn the style of speech from a specific character using this method of training.