mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
21 lines
446 B
Python
21 lines
446 B
Python
from collections import OrderedDict
|
|
import gc
|
|
import torch
|
|
from jobs.process import BaseExtensionProcess
|
|
|
|
|
|
def flush():
|
|
torch.cuda.empty_cache()
|
|
gc.collect()
|
|
|
|
|
|
class DatasetTools(BaseExtensionProcess):
|
|
|
|
def __init__(self, process_id: int, job, config: OrderedDict):
|
|
super().__init__(process_id, job, config)
|
|
|
|
def run(self):
|
|
super().run()
|
|
|
|
raise NotImplementedError("This extension is not yet implemented")
|