Skip to content

Instantly share code, notes, and snippets.

View littlewine's full-sized avatar

Antonios Minas Krasakis littlewine

View GitHub Profile
View train.py
import logging
import time
import argparse
import pandas as pd
import sklearn
import torch
from tqdm.auto import tqdm
from simpletransformers.retrieval import RetrievalModel, RetrievalArgs