storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Shell script to minimally test supported models on a real dataset (#204)

Browse Source 
* start minimal complete tests

* updates

* fixes

* fixes
This commit is contained in:
Sayak Paul
2025-01-10 07:22:00 +05:30
committed by GitHub
parent b9a6492ad1
commit 9523a05220
5 changed files with 295 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_dataset.py
+3 -3
View File
@@ -4,7 +4,7 @@ import sys
def test_video_dataset():
from dataset import VideoDataset
from cogvideox.dataset import VideoDataset
dataset_dirs = VideoDataset(
data_root="assets/tests/",
@@ -33,7 +33,7 @@ def test_video_dataset():
def test_video_dataset_with_resizing():
from dataset import VideoDatasetWithResizing
from cogvideox.dataset import VideoDatasetWithResizing
dataset_dirs = VideoDatasetWithResizing(
data_root="assets/tests/",
@@ -63,7 +63,7 @@ def test_video_dataset_with_resizing():
def test_video_dataset_with_bucket_sampler():
import torch
from dataset import BucketSampler, VideoDatasetWithResizing
from cogvideox.dataset import BucketSampler, VideoDatasetWithResizing
from torch.utils.data import DataLoader
dataset_dirs = VideoDatasetWithResizing(