14 lines
285 B
Python
14 lines
285 B
Python
|
from lib.train.admin.environment import env_settings
|
||
|
|
|
||
|
|
|
||
|
|
class Settings:
|
||
|
|
""" Training settings, e.g. the paths to datasets and networks."""
|
||
|
|
def __init__(self):
|
||
|
|
self.set_default()
|
||
|
|
|
||
|
|
def set_default(self):
|
||
|
|
self.env = env_settings()
|
||
|
|
self.use_gpu = True
|
||
|
|
|
||
|
|
|