search

AliaksandrSiarohin / first-order-model

This repository contains the source code for the paper First Order Motion Model for Image Animation
https://aliaksandrsiarohin.github.io/first-order-model-website/
MIT License
14.44k stars 3.2k forks source link

Trouble retraining with a GeForce® GTX 1660 Ti graphics card #442

Open PabloUpc2 opened 3 years ago

PabloUpc2 commented 3 years ago

Hi, the truth is that I am very new to the field of deep learning, so I have had a lot of trouble being able to retrain the model from its pre-trained model with the VoxCeleb dataset. I only have a GeForce® GTX 1660 Ti video card and even though I use only 8GB of the data set, the training process never progresses. What server (AWS, Azure, etc) option do you recommend for the training task? And what would be the approximate price that it would cost me to be able to retrain the model from its pretrained model with the VoxCeleb dataset? The truth is, I don't have much money and I would like to opt for this option as long as it is safe and not so expensive.

AliaksandrSiarohin commented 3 years ago

Did you adjust a batch size? You should set batch size to 5 or even less.