[example] opt does not depend on Titans (#1811)

This commit is contained in:
Jiarui Fang
2022-11-08 12:02:20 +08:00
committed by GitHub
parent 6fa71d65d3
commit 350ccc0481
4 changed files with 34 additions and 11 deletions

View File

@@ -32,9 +32,9 @@ import datasets
import torch
import torch.distributed as dist
from accelerate.utils import set_seed
from context import barrier_context
from datasets import load_dataset
from packaging import version
from titans.utils import barrier_context
from torch.utils.data import DataLoader
from tqdm.auto import tqdm
from utils import colo_memory_cap