修改dataloader
This commit is contained in:
+1
-1
@@ -13,7 +13,7 @@ import os
|
|||||||
from PIL import Image
|
from PIL import Image
|
||||||
import matplotlib.pyplot as plt
|
import matplotlib.pyplot as plt
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
def create_dataloaders(data_root='..',
|
def create_dataloaders(data_root='..',
|
||||||
batch_size=32,
|
batch_size=32,
|
||||||
|
|||||||
Reference in New Issue
Block a user