add some comments
This commit is contained in:
parent
2cad997a27
commit
37fdde8e83
12
.gitignore
vendored
12
.gitignore
vendored
@ -1,8 +1,14 @@
|
||||
./flowers/*
|
||||
.DS_Store
|
||||
./UNet/train_image/*
|
||||
./UNet/params/*
|
||||
./UNet/__pycache__/*
|
||||
UNet/train_image/*
|
||||
UNet/params/*
|
||||
UNet/__pycache__/*
|
||||
Logs_Checkpoints/
|
||||
data/
|
||||
archive.zip
|
||||
flowers/*
|
||||
VOCdevkit/
|
||||
inference_results
|
||||
.train.py.swp
|
||||
VOCdevkit_08-Jun-2007.tar
|
||||
wget-log
|
||||
|
@ -7,9 +7,9 @@ from net import *
|
||||
from torchvision.utils import save_image
|
||||
|
||||
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
||||
weight_path = r'/Users/hanzhangma/Nextcloud/mhz/Study/SS24/MasterThesis/UNet/params/unet.pth'
|
||||
data_path = r'/Users/hanzhangma/Document/DataSet/VOC2007'
|
||||
save_path = r'/Users/hanzhangma/Nextcloud/mhz/Study/SS24/MasterThesis/Unet/train_image'
|
||||
weight_path = r'/home/stud/hanzhang/MasterThesis/UNet/params/unet.pth'
|
||||
data_path = r'/home/stud/hanzhang/MasterThesis/VOCdevkit/VOCdevkit/VOC2007'
|
||||
save_path = r'/home/stud/hanzhang/MasterThesis/UNet/train_image'
|
||||
|
||||
if __name__ == '__main__':
|
||||
data_loader = DataLoader(MyDataset(data_path), batch_size= 4, shuffle=True)
|
||||
|
1616
diffusion.ipynb
1616
diffusion.ipynb
File diff suppressed because one or more lines are too long
Loading…
Reference in New Issue
Block a user