Ваш сеанс потерпел крах после использования всей доступной оперативной памяти в Google Colab - PullRequest
Новая
       18

Ваш сеанс потерпел крах после использования всей доступной оперативной памяти в Google Colab

0 голосов
/ 30 мая 2019

Я пытаюсь заставить это работать в Google Colab https://github.com/oawiles/X2Face/blob/master/UnwrapMosaic/Face2Face_UnwrapMosaic.ipynb,

Мне удалось запустить его и отобразить результаты один раз, но с тех пор в последнее время я получаю сбой вашей сессии после использования всей доступной оперативной памяти в этой ячейке 

    BASE_MODEL = '/scratch/shared/slow/ow/eccv/2018/release_models/' # Change to your path
state_dict = torch.load(BASE_MODEL + 'x2face_model.pth')

model = UnwrappedFaceWeightedAverage(output_num_channels=2, input_num_channels=3, inner_nc=128)
model.load_state_dict(state_dict['state_dict'])
model = model.cuda()

model = model.eval()

я не уверен, является ли это проблемой с колабом или нет

вот логи 

warn("IPython.utils.traitlets has moved to a top-level traitlets package.")
/usr/local/lib/python2.7/dist-packages/IPython/utils/traitlets.py:5: UserWarning: IPython.utils.traitlets has moved to a top-level traitlets package.
...