Update main.py
Browse files
main.py
CHANGED
|
@@ -27,7 +27,6 @@ class ManualCaptionDataset(Dataset):
|
|
| 27 |
self.transform = transform
|
| 28 |
self.captions = []
|
| 29 |
|
| 30 |
-
# Introducem manual descrierile pentru fiecare imagine
|
| 31 |
for img_name in self.img_names:
|
| 32 |
caption = 'Photo of Andrei smiling and dressed in winter clothes at a Christmas market'
|
| 33 |
self.captions.append(caption)
|
|
@@ -79,7 +78,7 @@ unet.train()
|
|
| 79 |
text_model.train()
|
| 80 |
|
| 81 |
# Definire număr de epoci
|
| 82 |
-
num_epochs = 5
|
| 83 |
|
| 84 |
# Training loop
|
| 85 |
for epoch in range(num_epochs):
|
|
|
|
| 27 |
self.transform = transform
|
| 28 |
self.captions = []
|
| 29 |
|
|
|
|
| 30 |
for img_name in self.img_names:
|
| 31 |
caption = 'Photo of Andrei smiling and dressed in winter clothes at a Christmas market'
|
| 32 |
self.captions.append(caption)
|
|
|
|
| 78 |
text_model.train()
|
| 79 |
|
| 80 |
# Definire număr de epoci
|
| 81 |
+
num_epochs = 5
|
| 82 |
|
| 83 |
# Training loop
|
| 84 |
for epoch in range(num_epochs):
|