<filename>save_resume_restart/pytorch_save_resume/utils.py<gh_stars>0
from __future__ import division
from __future__ import print_function


import logging

import torch

from polyaxon_helper import get_outputs_path

logging.basicConfig(level=logging.INFO)


def get_weight_filename():
    return '{}/{}'.format(get_outputs_path(), 'checkpoint.pth.tar')


def set_seed(seed, cuda):
    # Seed for reproducibility
    torch.manual_seed(seed)
    if cuda:
        torch.cuda.manual_seed(seed)

<filename>save_resume_restart/pytorch_save_resume/utils.py<gh_stars>0
from __future__ import division
from __future__ import print_function


import logging

import torch

from polyaxon_helper import get_outputs_path

logging.basicConfig(level=logging.INFO)


def get_weight_filename():
    return '{}/{}'.format(get_outputs_path(), 'checkpoint.pth.tar')


def set_seed(seed, cuda):
    # Seed for reproducibility
    torch.manual_seed(seed)
    if cuda:
        torch.cuda.manual_seed(seed)

# Seed for reproducibility