Jeremy Howard jph00
🦘
jph00
/ finetune_model.py
Created
August 6, 2023 04:55
— forked from conceptofmind/finetune_model.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| from datasets import load_dataset | |
| import argparse | |
| import os | |
| import math | |
| from itertools import chain | |
| from datetime import timedelta | |
| from torch.utils.data import DataLoader | |
| from accelerate import Accelerator | |
| from accelerate.utils import (DummyOptim, DummyScheduler, |