Exemplo n.º 1
0
# Date   : 3/17/17
# 
# This file is part of TensorArtist.

from tartist.app import gan
from tartist.app.gan import GANGraphKeys
from tartist.core import get_env, get_logger
from tartist.core.utils.naming import get_dump_directory, get_data_directory
from tartist.nn import opr as O, optimizer, summary

logger = get_logger(__file__)

__envs__ = {
    'dir': {
        'root': get_dump_directory(__file__),
        'data': get_data_directory('WellKnown/mnist')
    },

    'trainer': {
        'learning_rate': 0.001,

        'batch_size': 100,
        'epoch_size': 500,
        'nr_epochs': 100,

        'env_flags': {
            'log_device_placement': False
        }
    },
}
Exemplo n.º 2
0
# Date   : 2/27/17
#
# This file is part of TensorArtist.

from tartist.core import get_env, get_logger
from tartist.core.utils.naming import get_dump_directory, get_data_directory
from tartist.nn import opr as O, optimizer, summary

import functools

logger = get_logger(__file__)

__envs__ = {
    'dir': {
        'root': get_dump_directory(__file__),
        'data': get_data_directory('WellKnown/cifar')
    },
    'dataset': {
        'nr_classes': 10
    },
    'trainer': {
        'learning_rate': 0.01,
        'batch_size': 100,
        'epoch_size': 500,
        'nr_epochs': 100,
        'env_flags': {
            'log_device_placement': False
        }
    },
    'inference': {
        'batch_size': 256,
Exemplo n.º 3
0
# This file is part of TensorArtist.

import re

from tartist.app import gan
from tartist.app.gan import GANGraphKeys
from tartist.core import get_env, get_logger
from tartist.core.utils.naming import get_dump_directory, get_data_directory
from tartist.nn import opr as O, optimizer, summary

logger = get_logger(__file__)

__envs__ = {
    'dir': {
        'root': get_dump_directory(__file__),
        'data': get_data_directory('Pix2Pix/edges2faces')
    },
    'dataset': {
        'name': 'edges2shoes',
        'db_a': 'train_edges_db',
        'db_b': 'train_face_db',
    },
    'trainer': {
        'learning_rate': 2e-4,

        'batch_size': 64,
        'epoch_size': 1000,
        'nr_epochs': 200,
        'nr_g_per_iter': 1,
        'nr_d_per_iter': 1,
    }
# This file is part of TensorArtist.

import re

from tartist.app import gan
from tartist.app.gan import GANGraphKeys
from tartist.core import get_env, get_logger
from tartist.core.utils.naming import get_dump_directory, get_data_directory
from tartist.nn import opr as O, optimizer, summary

logger = get_logger(__file__)

__envs__ = {
    'dir': {
        'root': get_dump_directory(__file__),
        'data': get_data_directory('Pix2Pix/edges2shoes')
    },
    'dataset': {
        'name': 'edges2shoes',
        'db_a': 'train_edges_db',
        'db_b': 'train_shoes_db',
    },
    'trainer': {
        'learning_rate': 2e-4,
        'batch_size': 256,
        'epoch_size': 1000,
        'nr_epochs': 200,
        'nr_g_per_iter': 1,
        'nr_d_per_iter': 1,
    }
}
Exemplo n.º 5
0
# This file is part of TensorArtist.

import re

from tartist.app import gan
from tartist.app.gan import GANGraphKeys
from tartist.core import get_env, get_logger
from tartist.core.utils.naming import get_dump_directory, get_data_directory
from tartist.nn import opr as O, optimizer, summary

logger = get_logger(__file__)

__envs__ = {
    'dir': {
        'root': get_dump_directory(__file__),
        'data': get_data_directory('Pix2Pix/happy2surprise')
    },
    'dataset': {
        'name': 'happy2surprise',
        'db_a': 'train_happy_db',
        'db_b': 'train_surprise_db',
    },
    'trainer': {
        'learning_rate': 2e-4,
        'batch_size': 64,
        'epoch_size': 1000,
        'nr_epochs': 200,
        'nr_g_per_iter': 1,
        'nr_d_per_iter': 1,
    }
}