-
O trabalho foi feito no google colab, então é possível que alguns caminhos dos diretórios tenham que ser alterados, caso o notebook seja executado fora desse ambiente
-
Como mencionado no vídeo não é necessário treinar o modelo novamente. Basta substituir o caminho para o arquivo
model_epoch_140.pthna célula de hiperparâmetros:model.load_state_dict(torch.load('/content/drive/MyDrive/modelsNLP/model_epoch_140.pth'))
-
Notifications
You must be signed in to change notification settings - Fork 0
IgorRahzel/ImageCaptioning
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Implementing an Image Captioning model using a CNN as encoder and a RNN as decoder
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published