Skip to content

Instantly share code, notes, and snippets.

View weiqi-dyania's full-sized avatar

Weiqi Sun weiqi-dyania

View GitHub Profile
command_file: null
commands: null
compute_environment: LOCAL_MACHINE
deepspeed_config: {}
distributed_type: MULTI_GPU
downcast_bf16: 'no'
fsdp_config: {}
gpu_ids: all
machine_rank: 0
main_process_ip: null
"""
An example script to fine tune stanford/BioMedLM on dummy data without trainer and accelerate
"""
import math
import copy
import datasets
import torch
from datasets import Dataset, load_dataset
from torch.utils.data.dataloader import DataLoader
from tqdm.auto import tqdm