What do you want to save?
Add Code snippet
New code examples
-
Other 2021-09-20 06:26:07
Creating csv file
import pandas as pdimport osimport torchdevice = ("cuda" if torch.cuda.is_available() else "cpu")train_df = pd.DataFrame(columns=["img_name","label"])train_df["img_name"] = os.listdir("train/")fo... Add solution