AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-06-29 00:15:25 +02:00
Code Issues Projects Releases Packages Wiki Activity

Tests influence nb TF

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2019-11-08 16:50:02 -05:00
parent 0066da2e4d
commit 0d1a684aed
5 changed files with 46 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
higher/utils.py
View file

@ -43,6 +43,7 @@ def plot_res(log, fig_name='res'):
fig_name = fig_name.replace('.',',')
plt.savefig(fig_name)
plt.close()
def plot_compare(filenames, fig_name='res'):
@ -82,6 +83,7 @@ def plot_compare(filenames, fig_name='res'):
fig_name = fig_name.replace('.',',')
plt.savefig(fig_name, bbox_inches='tight')
plt.close()
def viz_sample_data(imgs, labels, fig_name='data_sample'):
@ -97,6 +99,7 @@ def viz_sample_data(imgs, labels, fig_name='data_sample'):
plt.xlabel(labels[i].item())
plt.savefig(fig_name)
plt.close()
def model_copy(src,dst, patch_copy=True, copy_grad=True):
#model=copy.deepcopy(fmodel) #Pas approprie, on ne souhaite que les poids/grad (pas tout fmodel et ses etats)