fastai--fastai
wehub-resource-sync
14 行
516 B
Python
14 行
516 B
Python
from torch import multiprocessing
|
|
import platform,os
|
|
if platform.system()=='Darwin':
|
|
# Python 3.8 changed to 'spawn' but that doesn't work with PyTorch DataLoader w n_workers>0
|
|
multiprocessing.set_start_method('fork', force=True)
|
|
# workaround "OMP: Error #15: Initializing libiomp5.dylib, but found libomp.dylib already initialized"
|
|
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
|
|
|
|
from .imports import *
|
|
from .torch_imports import *
|
|
from .torch_core import *
|
|
from .layers import *
|
|
from .losses import *
|