Skip to content

Commit

Permalink
Batch size optimization to reduce training time (#1638)
Browse files Browse the repository at this point in the history
  • Loading branch information
himanshugt16 authored Dec 12, 2024
1 parent 7a5aadb commit ebcdd78
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kairon/shared/llm/processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ async def train(self, user, *args, **kwargs) -> Dict:
await self.__delete_collections()
count = 0
processor = CognitionDataProcessor()
batch_size = 100
batch_size = 500

collections_data = CognitionData.objects(bot=self.bot)
collection_groups = {}
Expand Down

0 comments on commit ebcdd78

Please sign in to comment.