Model Saving

- The model is trained on the GPU if available. 5. **Saving the Model**: - After training, the fine-tuned model and tokenizer are saved to disk. ### Next Steps - **Evaluate the Model**: After training, evaluate the model on a valid…

query_embeddings = model(**query_encodings)['last_hidden_state'][:, 0, :] passage_embeddings = model(**passage_encodings)['last_hidden_state'][:, 0, :] # Apply dropout query_embeddings = dropout(query_embedd…

print(f"Early stopping triggered at epoch {epoch}") break print(f"Epoch {epoch+1}/{3000}, Training Loss: {loss.item():.4f}, Validation Loss: {avg_val_loss:.4f}") # Save the model torch.save(model.state_dict(), …

optimizer.step() # Zero gradients optimizer.zero_grad() # Validation loop scorer.eval() val_losses = [] with torch.no_grad(): for batch_inputs, batch_targets in val_loader: outpu…

redlock = Redlock([{"host": "localhost", "port": 6379, "db": 0}]) def save_model(version, data): lock_name = f"model_{version}_lock" lock = redlock.lock(lock_name, 10000) # Lock duration in milliseconds if not l…

reformulated_queries = [model.generate(tokenizer(f"reformulate: {q}", return_tensors="pt", max_length=512, truncation=True)['input_ids'], max_length=512)[0] for q in original_queries] reformulated_texts = [tokenizer.decode(output, skip_spec…