update train file

src/gptneo_cosmos.py

@@ -34,7 +34,7 @@ logger.setLevel(logging.INFO)
 tokenizer=GPT2Tokenizer.from_pretrained('EleutherAI/gpt-neo-1.3B',pad_token='<|endoftext|>')
 
 dataset=load_dataset('cosmos_qa')
-num_choices=2
+num_choices=4
 
 def preprocess(example):
     example['context&question']=example['context']+example['question']
@@ -232,4 +232,4 @@ if jax.process_index() == 0:
             push_to_hub=True,
             commit_message=f"Cosmos:Saving weights of epoch {epoch} at step {idx}",)
 
-summary_writer.flush()
+summary_writer.flush()