ImageCaptioning.pytorch icon indicating copy to clipboard operation
ImageCaptioning.pytorch copied to clipboard

verified publisher icon deep-learning-with-pytorch
→

Metadata

image captioning codebase in pytorch(finetunable cnn in branch "with_finetune";diverse beam search can be found in 'dbs' branch; self-critical training is under my self-critical.pytorch repository.)

Results 2 ImageCaptioning.pytorch issues
Sort by recently updated
recently updated
newest added

Make FCModel.py working on CUDA

12 comment

I tried to run `python eval.py --model ./data/FC/fc-model.pth --infos_path ./data/FC/fc-infos.pkl --image_folder ./data`(as said in the book on page 34) with different versions of PyTorch CUDA-enabled builds but all my attempts...

pbelevich avatar pbelevich

added check for GPU into eval.py

When running eval.py on Cuda 11 / pytorch 1.9 , ran into following error `RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but...

johnrey1 avatar johnrey1

← Metadata

89
Stars
56
Forks
Watchers

Owner

verified publisher icon deep-learning-with-pytorch

Metadata

image captioning codebase in pytorch(finetunable cnn in branch "with_finetune";diverse beam search can be found in 'dbs' branch; self-critical training is under my self-critical.pytorch repository.)

Back